diff --git a/tests/python/unittest/test_profiler.py b/tests/python/unittest/test_profiler.py index d58c08b5f26e..8e8209f5235b 100644 --- a/tests/python/unittest/test_profiler.py +++ b/tests/python/unittest/test_profiler.py @@ -112,7 +112,6 @@ def get_log(): var.asnumpy() stop = time.time() task.stop() - assert stop > start profiler.set_state('stop')