adds default value for is_train

2025-11-09 12:24:28 +01:00
parent c2b96a0c19
commit 51b18f1d82
1 changed files with 1 additions and 1 deletions
--- a/mmp/a4/dataset.py
+++ b/mmp/a4/dataset.py
@@ -90,7 +90,7 @@ def get_dataloader(
    num_workers: int,
    anchor_grid: np.ndarray,
    is_test: bool,
-    is_train: bool,
+    is_train: bool = False,
 ) -> DataLoader:
    dataset = MMP_Dataset(
        path_to_data=path_to_data,