Enhance error handling for current model version retrieval

safoinme · safoinme · commit 692798eb3642 · 2024-11-27T11:50:31.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
@@ -66,7 +66,7 @@ def compute_performance_metrics_on_current_data(
     try:
         current_version = Model(name=latest_version.name, version=target_env)
         current_version_number = current_version.number
-    except KeyError:
+    except (RuntimeError, KeyError):
         current_version_number = None
 
     if current_version_number is None:
