amend

pytorch · vmoens · Oct 2, 2023 · Sep 15, 2023 · Sep 15, 2023 · Sep 17, 2023
commit c789e50088759497d73cbb0101ba81472f96ee59
diff --git a/torchrl/envs/gym_like.py b/torchrl/envs/gym_like.py
@@ -247,6 +247,8 @@ def _step(self, tensordict: TensorDictBase) -> TensorDictBase:
         tensordict_out = TensorDict(obs_dict, batch_size=tensordict.batch_size)
 
         if self.info_dict_reader and info is not None:
+            if not isinstance(info, dict):
+                raise ValueError(f"Expected info to be a dictionary but got a {type(info)} with values {info}.")
             for info_dict_reader in self.info_dict_reader:
                 out = info_dict_reader(info, tensordict_out)
                 if out is not None: