adds default value for is_train

This commit is contained in:
franksim
2025-11-09 12:24:28 +01:00
parent c2b96a0c19
commit 51b18f1d82

View File

@@ -90,7 +90,7 @@ def get_dataloader(
num_workers: int, num_workers: int,
anchor_grid: np.ndarray, anchor_grid: np.ndarray,
is_test: bool, is_test: bool,
is_train: bool, is_train: bool = False,
) -> DataLoader: ) -> DataLoader:
dataset = MMP_Dataset( dataset = MMP_Dataset(
path_to_data=path_to_data, path_to_data=path_to_data,