amend

pytorch · Nov 22, 2023 · 2a4658e · 2a4658e
1 parent 80ee04f
commit 2a4658e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensordict/utils.py b/tensordict/utils.py
@@ -1726,6 +1726,8 @@ def _legacy_lazy(func):
 # Process initializer for map
 def _proc_init(base_seed, queue):
     print('init worker', os.getpid())
+    if queue.empty():
+        exit()
     worker_id = queue.get(timeout=10)
     print('worker id', worker_id)
     seed = base_seed + worker_id