isort/black

GFNOrg · Nov 14, 2024 · 98579e8 · 98579e8
1 parent 0adf808
commit 98579e8
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/gfn/samplers.py b/src/gfn/samplers.py
@@ -35,11 +35,7 @@ def sample_actions(
         save_estimator_outputs: bool = False,
         save_logprobs: bool = True,
         **policy_kwargs: Any,
-    ) -> Tuple[
-        Actions,
-        torch.Tensor | None,
-        torch.Tensor | None,
-    ]:
+    ) -> Tuple[Actions, torch.Tensor | None, torch.Tensor | None]:
         """Samples actions from the given states.
 
         Args: