From ec6968f86ec02781f7c093181e9c56397ae40fa6 Mon Sep 17 00:00:00 2001
From: archana-ramalingam <archana.ramalingam@amd.com>
Date: Wed, 2 Oct 2024 01:14:57 -0500
Subject: [PATCH] Remove unit tests for nightly

---
 sharktank/tests/evaluate/perplexity_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sharktank/tests/evaluate/perplexity_test.py b/sharktank/tests/evaluate/perplexity_test.py
index 3351f1aaf..e035e55eb 100644
--- a/sharktank/tests/evaluate/perplexity_test.py
+++ b/sharktank/tests/evaluate/perplexity_test.py
@@ -15,7 +15,7 @@
 logger.setLevel(logging.INFO)
 
 
-class PerplexityTest(unittest.TestCase):
+class PerplexityTest:
     @pytest.mark.expensive
     @pytest.mark.integration
     def test(self):
@@ -49,4 +49,4 @@ def test(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    PerplexityTest().test()