fix integration test

DeepRank · Jan 3, 2024 · 7f33a68 · 7f33a68
1 parent 3d31c59
commit 7f33a68
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -243,8 +243,7 @@ def test_nan_loss_cases(validate, best_model, hdf5_files_for_nan):
     dataset_valid = GraphDataset(
         hdf5_path = hdf5_files_for_nan,
         subset = [mols[0]],
-        dataset_train=dataset_train,
-        train=False
+        train_source=dataset_train
         )
 
     trainer = Trainer(