From 51b18f1d828dd2d7c14718a03b82a3580b786114 Mon Sep 17 00:00:00 2001 From: franksim Date: Sun, 9 Nov 2025 12:24:28 +0100 Subject: [PATCH] adds default value for is_train --- mmp/a4/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmp/a4/dataset.py b/mmp/a4/dataset.py index d642866..5d466ca 100644 --- 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,