amend

vmoens · vmoens · commit 37dbc1a5245b · 2025-07-04T12:08:40.000+01:00
diff --git a/torchrl/envs/transforms/transforms.py b/torchrl/envs/transforms/transforms.py
@@ -6531,9 +6531,7 @@ def _reset_func(
                     #         *reset_val.keys(True)
                     #     )
                     # )
-                    tensordict_reset = reset_val.new_zeros(_reset.shape)
-                    print(f"tensordict_reset: {tensordict_reset}")
-                    print(f"reset_val: {reset_val}")
+                    tensordict_reset = reset_val.new_zeros(_reset.shape, empty_lazy=True)
                     tensordict_reset[_reset] = reset_val
                 else:
                     resets = self.default_value(reset=_reset)
