diff --git a/test/test_profiler.py b/test/test_profiler.py index eb6822d9311..8ffb17a1707 100644 --- a/test/test_profiler.py +++ b/test/test_profiler.py @@ -125,7 +125,7 @@ def train_worker(): p.start() training_started.wait(60) # Delay to allow the profile to capture - time.sleep(5) + time.sleep(10) p.terminate() path = self._check_xspace_pb_exist(logdir) self._check_trace_namespace_exists(path)