amend

pytorch · vmoens · Jun 9, 2023 · Jun 9, 2023 · Jun 9, 2023 · Jun 9, 2023
commit 06937d44a170889e0f0c700271b4c820eef6f79f
diff --git a/torchrl/envs/vec_env.py b/torchrl/envs/vec_env.py
@@ -758,7 +758,7 @@ def _step(self, tensordict: TensorDictBase) -> TensorDictBase:
         # this is faster than update_ but won't work for lazy stacks
         for key in self.env_input_keys:
             self.shared_tensordict_parent._set(
-                tensordict.get(key), inplace=True,
+                key, tensordict.get(key), inplace=True,
             )
         if self.event is not None:
             self.event.record()