diff --git a/python/python/tests/test_torch.py b/python/python/tests/test_torch.py index 09a3c8de9..0c6e85294 100644 --- a/python/python/tests/test_torch.py +++ b/python/python/tests/test_torch.py @@ -325,6 +325,7 @@ def _fork_dataloader_diagnostics(worker_id=None): print( f"DataLoader pid={os.getpid()} worker={worker_id}", file=sys.stderr, flush=True ) + faulthandler.disable() faulthandler.enable(all_threads=True) faulthandler.register(signal.SIGUSR1, all_threads=True, chain=False)