Refactor exception handling for current version retrieval in performa…

…nce metrics computation
zenml-io · Nov 27, 2024 · 5c29e59 · 5c29e59
1 parent 4b61d49
commit 5c29e59
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 (RuntimeError, KeyError):
+    except KeyError:
         pass
 
     if current_version_number is None: