From 6012633f9ea430d5fc422ea10485568b44109597 Mon Sep 17 00:00:00 2001 From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com> Date: Mon, 2 Oct 2023 08:53:01 +0200 Subject: [PATCH 1/4] fix ExternalArtifact behavior --- template/pipelines/batch_inference.py | 2 +- template/pipelines/training.py | 2 +- template/run.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/template/pipelines/batch_inference.py b/template/pipelines/batch_inference.py index b19c3a7..8f9c26c 100644 --- a/template/pipelines/batch_inference.py +++ b/template/pipelines/batch_inference.py @@ -20,7 +20,7 @@ mlflow_model_registry_deployer_step, ) from zenml.logger import get_logger -from zenml.steps.external_artifact import ExternalArtifact +from zenml.artifacts.external_artifact import ExternalArtifact logger = get_logger(__name__) diff --git a/template/pipelines/training.py b/template/pipelines/training.py index 8939914..d80083d 100644 --- a/template/pipelines/training.py +++ b/template/pipelines/training.py @@ -31,7 +31,7 @@ ) from zenml.integrations.mlflow.steps.mlflow_registry import mlflow_register_model_step from zenml.logger import get_logger -from zenml.steps.external_artifact import ExternalArtifact +from zenml.artifacts.external_artifact import ExternalArtifact logger = get_logger(__name__) diff --git a/template/run.py b/template/run.py index 562fb91..49fc089 100644 --- a/template/run.py +++ b/template/run.py @@ -1,7 +1,7 @@ # {% include 'template/license_header' %} -from zenml.steps.external_artifact import ExternalArtifact +from zenml.artifacts.external_artifact import ExternalArtifact from zenml.logger import get_logger from pipelines import {{product_name}}_batch_inference, {{product_name}}_training from config import MetaConfig @@ -177,7 +177,7 @@ def main( logger.info( "Batch inference pipeline finished successfully! " "You can find predictions in Artifact Store using ID: " - f"`{str(artifact.upload_if_necessary())}`." + f"`{str(artifact.id)}`." ) From 6185729058589fea199ed7dc3d624f12de3a846f Mon Sep 17 00:00:00 2001 From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com> Date: Mon, 2 Oct 2023 08:54:24 +0200 Subject: [PATCH 2/4] update zenml-ref --- .github/workflows/ci.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 0a81ca4..7de3135 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -35,3 +35,4 @@ jobs: with: stack-name: ${{ matrix.stack-name }} python-version: ${{ matrix.python-version }} + zenml-ref: feature/OSS-2424-extend-external-artifact From 8ae5d7bf736c8babc22aef6976bd6300adb0bebc Mon Sep 17 00:00:00 2001 From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com> Date: Mon, 2 Oct 2023 08:57:57 +0200 Subject: [PATCH 3/4] fix ref name --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 7de3135..d7e512c 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -35,4 +35,4 @@ jobs: with: stack-name: ${{ matrix.stack-name }} python-version: ${{ matrix.python-version }} - zenml-ref: feature/OSS-2424-extend-external-artifact + ref-zenml: feature/OSS-2424-extend-external-artifact From 9f5ebfbe6f7ed207f4f541f0eb55d25a45a98032 Mon Sep 17 00:00:00 2001 From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com> Date: Tue, 10 Oct 2023 15:52:58 +0200 Subject: [PATCH 4/4] update ref branch --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index d7e512c..3dbe72b 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -35,4 +35,4 @@ jobs: with: stack-name: ${{ matrix.stack-name }} python-version: ${{ matrix.python-version }} - ref-zenml: feature/OSS-2424-extend-external-artifact + ref-zenml: feature/OSS-2300-model-watch-tower-v0.1