rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-10-27 14:06:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(python): restore fatal traceback handler after torch worker setup

Browse Source
This commit is contained in:
Xuanwo
2026-10-27 02:05:05 +08:00
parent 1ce2427478
commit 23658d992c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/python/tests/test_torch.py
+1
View File
@@ -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)