chore: black

jspaezp · Nov 22, 2024 · fadbd95 · fadbd95
1 parent d16deeb
commit fadbd95
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_training.py b/tests/test_training.py
@@ -52,9 +52,9 @@ def test_base_train_works_on_schdulers(datamodule, scheduler, tiny_model):
 def mod_train_with_missing(datadir):
     mod = model.PepTransformerModel(nhead=4, d_model=64)
     datamodule = datamodules.PeptideDataModule(batch_size=5, base_dir=datadir)
-    datamodule.train_df.loc[[x for x in range(len(datamodule.train_df))], "mIRT"] = (
-        np.nan
-    )
+    datamodule.train_df.loc[
+        [x for x in range(len(datamodule.train_df))], "mIRT"
+    ] = np.nan
     datamodule.val_df.loc[[x for x in range(len(datamodule.val_df))], "mIRT"] = np.nan
     datamodule.setup()