diff --git a/gflownet/utils/buffer.py b/gflownet/utils/buffer.py index b2f41ac90..1593c48ee 100644 --- 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