Enhance error handling for current model version retrieval

zenml-io · Nov 27, 2024 · 692798e · 692798e
1 parent 74baad4
commit 692798e
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
@@ -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: