Refactor exception handling in performance metrics computation to cat…

…ch all exceptions
zenml-io · Nov 27, 2024 · 0b09eb6 · 0b09eb6
1 parent 5c29e59
commit 0b09eb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_and_deploy/steps/promotion/compute_performance_metrics_on_current_data.py b/train_and_deploy/steps/promotion/compute_performance_metrics_on_current_data.py
@@ -65,7 +65,7 @@ def compute_performance_metrics_on_current_data(
     current_version_number = None
     try:
         current_version = Model(name=latest_version.name, version=ModelStages.STAGING)
-    except KeyError:
+    except Exception:
         pass
 
     if current_version_number is None: