fixed config dataset_type validating

luxonis · Aug 20, 2024 · b71d5c8 · b71d5c8
1 parent 672ad7d
commit b71d5c8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/luxonis_train/utils/config.py b/luxonis_train/utils/config.py
@@ -173,6 +173,7 @@ def validate_params(self) -> Self:
         dataset_type = self.params.get("dataset_type")
         if dataset_type is None:
             return self
+        dataset_type = dataset_type.upper()
 
         if dataset_type not in DatasetType.__members__:
             raise ValueError(