[CI] fix UT success check in test_eagle_infer_beta_dp_attention.py (#14831)
This commit is contained in:
@@ -98,11 +98,4 @@ class TestEagleDPAttnServerLarge(CustomTestCase):
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
s = unittest.TestSuite()
|
||||
large_test = unittest.defaultTestLoader.loadTestsFromTestCase(
|
||||
TestEagleDPAttnServerLarge
|
||||
)
|
||||
s.addTest(large_test)
|
||||
|
||||
runner = unittest.TextTestRunner()
|
||||
runner.run(s)
|
||||
unittest.main()
|
||||
|
||||
@@ -80,11 +80,4 @@ class TestEagleDPAttnServerSmall(CustomTestCase):
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
s = unittest.TestSuite()
|
||||
small_test = unittest.defaultTestLoader.loadTestsFromTestCase(
|
||||
TestEagleDPAttnServerSmall
|
||||
)
|
||||
s.addTest(small_test)
|
||||
|
||||
runner = unittest.TextTestRunner()
|
||||
runner.run(s)
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user