formatting

alexhernandezgarcia · Sep 23, 2024 · 595db80 · 595db80
1 parent 9395e61
commit 595db80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gflownet/policy/base.py b/gflownet/policy/base.py
@@ -54,9 +54,9 @@ def __init__(
         # Checkpoint, defaults to None
         self.checkpoint = config.get("checkpoint", None)
         # TODO: This could be done better? We could store this only when using CNN policy. e.g. self.type could be "cnn"
-        if hasattr(env, 'height'):
+        if hasattr(env, "height"):
             self.height = env.height
-        if hasattr(env, 'width'):
+        if hasattr(env, "width"):
             self.width = env.width
         # Instantiate the model
         self.model, self.is_model = self.make_model()