fix: adjust to latest format

recursionpharma · Mar 25, 2024 · a0d01e7 · a0d01e7
1 parent 9bf35cd
commit a0d01e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/gflownet/algo/trajectory_balance.py b/src/gflownet/algo/trajectory_balance.py
@@ -342,7 +342,10 @@ def construct_batch(self, trajs, cond_info, log_rewards):
         return batch
 
     def compute_batch_losses(
-        self, model: TrajectoryBalanceModel, batch: gd.Batch, num_bootstrap: int = 0  # type: ignore[override]
+        self,
+        model: TrajectoryBalanceModel,
+        batch: gd.Batch,
+        num_bootstrap: int = 0,  # type: ignore[override]
     ):
         """Compute the losses over trajectories contained in the batch
 

diff --git a/src/gflownet/data/qm9.py b/src/gflownet/data/qm9.py
@@ -14,7 +14,6 @@
 class QM9Dataset(Dataset):
     def __init__(self, h5_file=None, xyz_file=None, train=True, targets=["gap"], split_seed=142857, ratio=0.9):
         if h5_file is not None:
-
             self.hdf = pd.HDFStore(h5_file, "r")
             self.df = self.hdf["df"]
             self.is_hdf = True