[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__":
|
if __name__ == "__main__":
|
||||||
s = unittest.TestSuite()
|
unittest.main()
|
||||||
large_test = unittest.defaultTestLoader.loadTestsFromTestCase(
|
|
||||||
TestEagleDPAttnServerLarge
|
|
||||||
)
|
|
||||||
s.addTest(large_test)
|
|
||||||
|
|
||||||
runner = unittest.TextTestRunner()
|
|
||||||
runner.run(s)
|
|
||||||
|
|||||||
@@ -80,11 +80,4 @@ class TestEagleDPAttnServerSmall(CustomTestCase):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
s = unittest.TestSuite()
|
unittest.main()
|
||||||
small_test = unittest.defaultTestLoader.loadTestsFromTestCase(
|
|
||||||
TestEagleDPAttnServerSmall
|
|
||||||
)
|
|
||||||
s.addTest(small_test)
|
|
||||||
|
|
||||||
runner = unittest.TextTestRunner()
|
|
||||||
runner.run(s)
|
|
||||||
|
|||||||
Reference in New Issue
Block a user