patch: Raise an informative error in case of unexpected task_name

IGNF · Aug 8, 2023 · b98f120 · b98f120
1 parent 3047670
commit b98f120
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,8 @@
 # main
 
+# 3.4.8
+- Raise an informative error in case of unexpected task_name.
+
 # 3.4.7
 - Remove tqdm when splitting a lidar tile to avoid cluttered logs during data preparation.
 

diff --git a/package_metadata.yaml b/package_metadata.yaml
@@ -1,4 +1,4 @@
-__version__: "3.4.7"
+__version__: "3.4.8"
 __name__: "myria3d"
 __url__: "https://github.com/IGNF/myria3d"
 __description__: "Deep Learning for the Semantic Segmentation of Aerial Lidar Point Clouds"

diff --git a/run.py b/run.py
@@ -128,6 +128,7 @@ def launch_hdf5(config: DictConfig):
         launch_hdf5()
 
     else:
-        log.warning(
-            f"Task {task_name} is not known. Specify a valid task name via task.task_name=..."
+        choices = ", ".join(task.value for task in TASK_NAMES)
+        raise ValueError(
+            f"Task '{task_name}' is not known. Specify a valid task name via task.task_name. Valid choices are: {choices})"
         )