### What this PR does / why we need it? Revert PR 5253 to fix the smoking problem ### Does this PR introduce _any_ user-facing change? Does not. ### How was this patch tested? It was tested in the failure case. Signed-off-by: Rifa <865071616@qq.com>
This commit is contained in:
@@ -1020,6 +1020,8 @@ class TestNPUWorker(TestBase):
|
||||
# Verify _dummy_run call count and order (by size descending)
|
||||
expected_calls = [
|
||||
unittest.mock.call(16),
|
||||
unittest.mock.call(8),
|
||||
unittest.mock.call(4),
|
||||
unittest.mock.call(1),
|
||||
]
|
||||
worker.model_runner._dummy_run.assert_has_calls(expected_calls)
|
||||
@@ -1028,7 +1030,7 @@ class TestNPUWorker(TestBase):
|
||||
worker.model_runner.capture_model.assert_not_called()
|
||||
|
||||
# Verify log output
|
||||
self.assertEqual(mock_logger.info.call_count, 2)
|
||||
self.assertEqual(mock_logger.info.call_count, 4)
|
||||
|
||||
# Verify seed setting
|
||||
mock_seed_everything.assert_called_once_with(12345)
|
||||
|
||||
Reference in New Issue
Block a user