lint

pytorch · vmoens · Oct 24, 2023 · Oct 6, 2023 · Oct 6, 2023 · Oct 6, 2023
commit b7b2081af094d948efb8a52855e5b44ce627e552
diff --git a/torchrl/envs/libs/pettingzoo.py b/torchrl/envs/libs/pettingzoo.py
@@ -482,8 +482,10 @@ def _reset(
 
         _reset = tensordict.get("_reset", None)
         if _reset is not None and not _reset.all():
-            raise RuntimeError(f"An attempt to call {type(self)}._reset was made when no reset signal could be found. "
-                               f"Expected '_reset' entry to be `tensor(True)` or `None` but got `{_reset}`.")
+            raise RuntimeError(
+                f"An attempt to call {type(self)}._reset was made when no reset signal could be found. "
+                f"Expected '_reset' entry to be `tensor(True)` or `None` but got `{_reset}`."
+            )
         if self.parallel:
             # This resets when any is done
             observation_dict, info_dict = self._reset_parallel(**kwargs)

diff --git a/torchrl/modules/tensordict_module/actors.py b/torchrl/modules/tensordict_module/actors.py
@@ -1758,8 +1758,10 @@ def set_tensor_keys(self, **kwargs):
         self.action_key = action_key
         self.return_to_go_key = return_to_go_key
         if out_action_key not in self.td_module.out_keys:
-            raise ValueError(f"The value of out_action_key ({out_action_key}) must be "
-                             f"within the actor output keys ({self.td_module.out_keys}).")
+            raise ValueError(
+                f"The value of out_action_key ({out_action_key}) must be "
+                f"within the actor output keys ({self.td_module.out_keys})."
+            )
         self.out_action_key = out_action_key
 
     def step(self, frames: int = 1) -> None: