From 98579e81804f1825978eaf3ef2a93faaeb219a40 Mon Sep 17 00:00:00 2001 From: Joseph Viviano Date: Thu, 14 Nov 2024 18:16:31 -0500 Subject: [PATCH] isort/black --- src/gfn/samplers.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/gfn/samplers.py b/src/gfn/samplers.py index 0df9449f..819620f0 100644 --- 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: