Buffer: Handle case where logger is None

alexhernandezgarcia · Jun 11, 2024 · f8c235c · f8c235c
1 parent 7b11c38
commit f8c235c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gflownet/utils/buffer.py b/gflownet/utils/buffer.py
@@ -28,7 +28,11 @@ def __init__(
         logger=None,
         **kwargs,
     ):
-        self.datadir = logger.datadir
+        if logger is not None:
+            self.datadir = logger.datadir
+        else:
+            self.datadir = Path("./logs")
+            self.datadir.mkdir(parents=True, exist_ok=True)
         self.env = env
         self.proxy = proxy
         self.replay_capacity = replay_capacity