diff --git a/template/steps/inference/inference_get_current_version.py b/template/steps/inference/inference_get_current_version.py index bdcecd3..0454d87 100644 --- a/template/steps/inference/inference_get_current_version.py +++ b/template/steps/inference/inference_get_current_version.py @@ -25,7 +25,6 @@ def inference_get_current_version() -> Annotated[str, "model_version"]: current_version = model_registry.list_model_versions( name=MetaConfig.mlflow_model_name, - metadata={}, stage=MetaConfig.target_env, )[0].version logger.info( diff --git a/template/steps/promotion/promote_get_versions.py b/template/steps/promotion/promote_get_versions.py index 879fda4..52d3529 100644 --- a/template/steps/promotion/promote_get_versions.py +++ b/template/steps/promotion/promote_get_versions.py @@ -32,7 +32,6 @@ def promote_get_versions() -> ( ### ADD YOUR OWN CODE HERE - THIS IS JUST AN EXAMPLE ### none_versions = model_registry.list_model_versions( name=MetaConfig.mlflow_model_name, - metadata={}, stage=None, ) latest_versions = none_versions[0].version @@ -40,7 +39,6 @@ def promote_get_versions() -> ( target_versions = model_registry.list_model_versions( name=MetaConfig.mlflow_model_name, - metadata={}, stage=MetaConfig.target_env, ) current_version = latest_versions diff --git a/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} b/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} index 00bd6e4..045f1f7 100644 --- a/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} +++ b/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} @@ -72,13 +72,11 @@ def promote_metric_compare_promoter( name=MetaConfig.mlflow_model_name, version=current_version, stage=ModelVersionStage.ARCHIVED, - metadata={}, ) model_registry.update_model_version( name=MetaConfig.mlflow_model_name, version=latest_version, stage=MetaConfig.target_env, - metadata={}, ) promoted_version = latest_version