Merge pull request #428 from Nota-NetsPresso/427-path-check-bug

Fix test directory check line
Nota-NetsPresso · Apr 22, 2024 · 7d4c0ae · 7d4c0ae
2 parents e405599 + 70dbe9b
commit 7d4c0ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,7 +6,7 @@
 
 ## Bug Fixes:
 
-No changes to highlight.
+- Fix test directory check line by `@illian01` in [PR 428](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/428)
 
 ## Breaking Changes:
 

diff --git a/src/netspresso_trainer/dataloaders/builder.py b/src/netspresso_trainer/dataloaders/builder.py
@@ -27,7 +27,7 @@ def dataset_path_check(conf_data: DictConfig, mode: Literal['train', 'test']):
         conf_data.path.test.label = None
 
     elif mode == 'test':
-        assert conf_data.path.train.test_dataset is not None, "For test, test split of dataset must be provided."
+        assert conf_data.path.test.image is not None, "For test, test split of dataset must be provided."
 
         if conf_data.path.train.image:
             logger.warning('For test (evaluation or inference), train split of dataset is not needed. This field will be ignored.')