Use traitlet's TraitError

pangeo-forge · Feb 20, 2024 · 20ebde6 · 20ebde6
1 parent 191955a
commit 20ebde6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pangeo_forge_runner/bakery/base.py b/pangeo_forge_runner/bakery/base.py
@@ -1,7 +1,7 @@
 from typing import List
 
 from apache_beam.pipeline import Pipeline, PipelineOptions
-from traitlets import Bool, TraitError
+from traitlets import TraitError
 from traitlets.config import LoggingConfigurable
 
 from ..commands.bake import Bake, ExecutionMetadata

diff --git a/pangeo_forge_runner/commands/bake.py b/pangeo_forge_runner/commands/bake.py
@@ -133,7 +133,7 @@ def _validate_job_name(self, proposal):
         """
         validating_regex = r"^[a-z][-_0-9a-z]{0,62}$"
         if not re.match(validating_regex, proposal.value):
-            raise ValueError(
+            raise TraitError(
                 f"job_name must match the regex {validating_regex}, instead found {proposal.value}"
             )
         return proposal.value