From 2a4658e8f37e7eefe1970c65b3ce80fa39868df3 Mon Sep 17 00:00:00 2001
From: vmoens <vincentmoens@gmail.com>
Date: Wed, 22 Nov 2023 16:03:29 +0000
Subject: [PATCH] amend

---
 tensordict/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensordict/utils.py b/tensordict/utils.py
index 8c19da010..87fba344b 100644
--- 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