Keep _SingleThreadedMapper to continue using a daemon thread

divyanshk · web-flow · commit 2a9eb5dde0c8 · 2025-05-13T17:30:44.000-07:00
Differential Revision: D74673949 Pull Request resolved: #1487
diff --git a/torchdata/nodes/map.py b/torchdata/nodes/map.py
@@ -573,7 +573,7 @@ def __init__(
                 self._sem,
                 self._stop_event,
             ),
-            daemon=False,
+            daemon=True,
             name=f"worker_thread(target={self.worker.__name__})",
         )
         self._thread.start()

Original file line number	Diff line number	Diff line change
`@@ -573,7 +573,7 @@ def __init__(`
`573`	`573`	`self._sem,`
`574`	`574`	`self._stop_event,`
`575`	`575`	`),`
`576`		`- daemon=False,`
	`576`	`+ daemon=True,`
`577`	`577`	`name=f"worker_thread(target={self.worker.__name__})",`
`578`	`578`	`)`
`579`	`579`	`self._thread.start()`