Remove unit tests for nightly

nod-ai · Oct 2, 2024 · ec6968f · ec6968f
1 parent 7e49580
commit ec6968f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()