black

GFNOrg · Mar 4, 2025 · 96394d7 · 96394d7
1 parent 2e9b035
commit 96394d7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/gfn/states.py b/src/gfn/states.py
@@ -84,7 +84,10 @@ def batch_shape(self, batch_shape: tuple[int, ...]) -> None:
 
     @classmethod
     def from_batch_shape(
-        cls, batch_shape: int | tuple[int, ...], random: bool = False, sink: bool = False
+        cls,
+        batch_shape: int | tuple[int, ...],
+        random: bool = False,
+        sink: bool = False,
     ) -> States | GraphStates:
         """Create a States object with the given batch shape.