minor edit on MetaTune class __init__.py method

ShiroDima · May 30, 2023 · ca85668 · ca85668
1 parent be6a2c1
commit ca85668
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/_metatune.py b/_metatune.py
@@ -72,7 +72,8 @@ def __init__(
 
         valid_tasks: Iterable[str] = ["classification", "regression"]
         if task not in valid_tasks:
-            raise ValueError(f"Invalid task {task}, expects tasks to be 'regression' or 'classification'")
+            raise ValueError(
+                f"Invalid task {task}, expects tasks to be 'regression' or 'classification', got {task}")
 
         self.task = task
         self.custom_tuners = custom_tuners