From 4edd5924c45a5969a8cd353087164d4516f34648 Mon Sep 17 00:00:00 2001
From: vmoens <vincentmoens@gmail.com>
Date: Wed, 22 Nov 2023 17:08:17 +0000
Subject: [PATCH] amend

---
 tensordict/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensordict/base.py b/tensordict/base.py
index f44fd76e1..8da51f258 100644
--- a/tensordict/base.py
+++ b/tensordict/base.py
@@ -3041,6 +3041,7 @@ def map(
         imap = pool.imap(fn, self_split, chunksize)
         while len(out) < len(self_split):
             out.append(imap.next(timeout=50.0))
+            print(len(out))
         # out = pool.map(fn, self_split, chunksize)
         out = torch.cat(list(imap), dim)
         return out