From 0b66f072abd9c5700e6f755a254b91dce1c8945c Mon Sep 17 00:00:00 2001 From: GitHub Actions Date: Mon, 21 Oct 2024 08:47:06 +0000 Subject: [PATCH 1/2] Auto-update of E2E template --- examples/e2e/.copier-answers.yml | 2 +- examples/e2e/steps/deployment/deployment_deploy.py | 2 +- examples/e2e/steps/hp_tuning/hp_tuning_select_best_model.py | 2 +- examples/e2e/steps/promotion/promote_with_metric_compare.py | 6 +++--- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/e2e/.copier-answers.yml b/examples/e2e/.copier-answers.yml index 74cc33d8594..b008b2c1e99 100644 --- a/examples/e2e/.copier-answers.yml +++ b/examples/e2e/.copier-answers.yml @@ -1,5 +1,5 @@ # Changes here will be overwritten by Copier -_commit: 2024.10.10 +_commit: 2024.10.21 _src_path: gh:zenml-io/template-e2e-batch data_quality_checks: true email: info@zenml.io diff --git a/examples/e2e/steps/deployment/deployment_deploy.py b/examples/e2e/steps/deployment/deployment_deploy.py index 3fb0d879f3f..dad351e45be 100644 --- a/examples/e2e/steps/deployment/deployment_deploy.py +++ b/examples/e2e/steps/deployment/deployment_deploy.py @@ -67,7 +67,7 @@ def deployment_deploy() -> ( registry_model_name=model.name, registry_model_version=model.run_metadata[ "model_registry_version" - ].value, + ], replace_existing=True, ) else: diff --git a/examples/e2e/steps/hp_tuning/hp_tuning_select_best_model.py b/examples/e2e/steps/hp_tuning/hp_tuning_select_best_model.py index 7d5a6bc33ea..65e524ecd98 100644 --- a/examples/e2e/steps/hp_tuning/hp_tuning_select_best_model.py +++ b/examples/e2e/steps/hp_tuning/hp_tuning_select_best_model.py @@ -50,7 +50,7 @@ def hp_tuning_select_best_model( hp_output = model.get_data_artifact("hp_result") model_: ClassifierMixin = hp_output.load() # fetch metadata we attached earlier - metric = float(hp_output.run_metadata["metric"].value) + metric = float(hp_output.run_metadata["metric"]) if best_model is None or best_metric < metric: best_model = model_ ### YOUR CODE ENDS HERE ### diff --git a/examples/e2e/steps/promotion/promote_with_metric_compare.py b/examples/e2e/steps/promotion/promote_with_metric_compare.py index 038d219d32d..6bc580f47ba 100644 --- a/examples/e2e/steps/promotion/promote_with_metric_compare.py +++ b/examples/e2e/steps/promotion/promote_with_metric_compare.py @@ -92,14 +92,14 @@ def promote_with_metric_compare( # Promote in Model Registry latest_version_model_registry_number = latest_version.run_metadata[ "model_registry_version" - ].value + ] if current_version_number is None: current_version_model_registry_number = ( latest_version_model_registry_number ) else: current_version_model_registry_number = ( - current_version.run_metadata["model_registry_version"].value + current_version.run_metadata["model_registry_version"] ) promote_in_model_registry( latest_version=latest_version_model_registry_number, @@ -111,7 +111,7 @@ def promote_with_metric_compare( else: promoted_version = current_version.run_metadata[ "model_registry_version" - ].value + ] logger.info( f"Current model version in `{target_env}` is `{promoted_version}` registered in Model Registry" From 4b2434aa0e72057803234e07de6325d6dca0bab9 Mon Sep 17 00:00:00 2001 From: GitHub Actions Date: Mon, 21 Oct 2024 08:50:05 +0000 Subject: [PATCH 2/2] Auto-update of NLP template --- examples/e2e_nlp/.copier-answers.yml | 2 +- examples/e2e_nlp/gradio/requirements.txt | 2 +- .../e2e_nlp/steps/promotion/promote_get_metrics.py | 12 ++++-------- 3 files changed, 6 insertions(+), 10 deletions(-) diff --git a/examples/e2e_nlp/.copier-answers.yml b/examples/e2e_nlp/.copier-answers.yml index 3ca2ba198fe..e509aae2760 100644 --- a/examples/e2e_nlp/.copier-answers.yml +++ b/examples/e2e_nlp/.copier-answers.yml @@ -1,5 +1,5 @@ # Changes here will be overwritten by Copier -_commit: 2024.09.23 +_commit: 2024.10.21 _src_path: gh:zenml-io/template-nlp accelerator: cpu cloud_of_choice: aws diff --git a/examples/e2e_nlp/gradio/requirements.txt b/examples/e2e_nlp/gradio/requirements.txt index 1bddfdfb85b..b53f1df9e62 100644 --- a/examples/e2e_nlp/gradio/requirements.txt +++ b/examples/e2e_nlp/gradio/requirements.txt @@ -9,4 +9,4 @@ pandas==1.5.3 session_info==1.0.0 scikit-learn==1.5.0 transformers==4.28.1 -IPython==7.34.0 \ No newline at end of file +IPython==8.10.0 \ No newline at end of file diff --git a/examples/e2e_nlp/steps/promotion/promote_get_metrics.py b/examples/e2e_nlp/steps/promotion/promote_get_metrics.py index 7f2951a5865..b24ac42245c 100644 --- a/examples/e2e_nlp/steps/promotion/promote_get_metrics.py +++ b/examples/e2e_nlp/steps/promotion/promote_get_metrics.py @@ -56,9 +56,7 @@ def promote_get_metrics() -> ( # Get current model version metric in current run model = get_step_context().model - current_metrics = ( - model.get_model_artifact("model").run_metadata["metrics"].value - ) + current_metrics = model.get_model_artifact("model").run_metadata["metrics"] logger.info(f"Current model version metrics are {current_metrics}") # Get latest saved model version metric in target environment @@ -72,11 +70,9 @@ def promote_get_metrics() -> ( except KeyError: latest_version = None if latest_version: - latest_metrics = ( - latest_version.get_model_artifact("model") - .run_metadata["metrics"] - .value - ) + latest_metrics = latest_version.get_model_artifact( + "model" + ).run_metadata["metrics"] logger.info(f"Latest model version metrics are {latest_metrics}") else: logger.info("No currently promoted model version found.")