Merge pull request #9 from Leon0402/fix/catchNonExistentDatasets

Raise exception for non-existent datasets.
NVlabs · Jul 17, 2022 · 37ad56f · 37ad56f
2 parents 9b6283c + f6a43e8
commit 37ad56f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/trajdata/utils/env_utils.py b/src/trajdata/utils/env_utils.py
@@ -27,6 +27,8 @@ def get_raw_dataset(dataset_name: str, data_dir: str) -> RawDataset:
 
     if "eupeds" in dataset_name:
         return EUPedsDataset(dataset_name, data_dir, parallelizable=True)
+
+    raise ValueError(f"Dataset with name '{dataset_name}' is not supported")
 
 
 def get_raw_datasets(data_dirs: Dict[str, str]) -> List[RawDataset]: