lint

pytorch · vmoens · Oct 5, 2023 · Jun 27, 2023 · Jun 27, 2023 · Jun 28, 2023
commit 87501ea32444de853b6c8563312beca30503b1c2
diff --git a/torchrl/data/rlhf/utils.py b/torchrl/data/rlhf/utils.py
@@ -15,6 +15,7 @@
 
 _has_transformers = importlib.util.find_spec("transformers") is not None
 
+
 class RolloutFromModel:
     """A class for performing rollouts with causal language models.
 
@@ -90,9 +91,9 @@ def __init__(
     ):
         if not _has_transformers:
             raise ImportError(
-            "transformers module couldn't be found. Make sure it is installed in your "
-            "environment."
-        )
+                "transformers module couldn't be found. Make sure it is installed in your "
+                "environment."
+            )
         self.model = model
         self.ref_model = ref_model
         self.reward_model = reward_model