amend

pytorch · vmoens · Jan 30, 2024 · Jan 25, 2024 · Jan 25, 2024 · Jan 25, 2024
commit 0d921c5eb5d097186d302ba80b665ccd53d0eb1a
diff --git a/torchrl/collectors/collectors.py b/torchrl/collectors/collectors.py
@@ -34,7 +34,6 @@
     TensorDict,
     TensorDictBase,
     TensorDictParams,
-    unravel_key,
 )
 from tensordict.nn import TensorDictModule, TensorDictModuleBase
 from tensordict.utils import NestedKey

diff --git a/torchrl/envs/batched_envs.py b/torchrl/envs/batched_envs.py
@@ -649,7 +649,6 @@ def start(self) -> None:
         self._start_workers()
 
     def to(self, device: DEVICE_TYPING):
-        print("super to")
         device = torch.device(device)
         if device == self.device:
             return self
@@ -865,7 +864,6 @@ def to(self, device: DEVICE_TYPING):
             return self
         super().to(device)
         if not self.is_closed:
-            print("casting envs")
             self._envs = [env.to(device) for env in self._envs]
         return self