Refactor exception handling for current version retrieval in performance metrics computation

safoinme · safoinme · commit 5c29e59340b3 · 2024-11-27T14:32:25.000+01:00
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:
