zenml-io · avishniakov · Nov 17, 2023 · Nov 10, 2023 · Nov 10, 2023 · Nov 10, 2023
diff --git a/.github/workflows/update-templates-to-examples.yml b/.github/workflows/update-templates-to-examples.yml
@@ -56,7 +56,7 @@ jobs:
           python-version: ${{ inputs.python-version }}
           stack-name: local
           ref-zenml: ${{ github.ref }}
-          ref-template: '2023.11.01' # Make sure it is aligned with ZENML_PROJECT_TEMPLATES from src/zenml/cli/base.py
+          ref-template: '2023.11.14' # Make sure it is aligned with ZENML_PROJECT_TEMPLATES from src/zenml/cli/base.py
 
       - name: Clean-up
         run: |
@@ -76,6 +76,7 @@ jobs:
 
       - name: Check-out fresh E2E template
         run: |
+          rm -rf examples/e2e
           mkdir -p examples/e2e
           printf '[email protected]' | zenml init --path examples/e2e --template e2e_batch --template-with-defaults
           bash scripts/format.sh

diff --git a/examples/e2e/.copier-answers.yml b/examples/e2e/.copier-answers.yml
@@ -1,5 +1,5 @@
 # Changes here will be overwritten by Copier
-_commit: 2023.11.01
+_commit: 2023.11.14
 _src_path: gh:zenml-io/template-e2e-batch
 data_quality_checks: true
 email: ''

diff --git a/examples/e2e/artifacts/__init__.py b/examples/e2e/artifacts/__init__.py
diff --git a/examples/e2e/artifacts/materializer.py b/examples/e2e/artifacts/materializer.py
diff --git a/examples/e2e/artifacts/model_metadata.py b/examples/e2e/artifacts/model_metadata.py
diff --git a/examples/e2e/config.py b/examples/e2e/config.py
diff --git a/examples/e2e/configs/deployer_config.yaml b/examples/e2e/configs/deployer_config.yaml
@@ -34,7 +34,7 @@ steps:
       notify_on_success: False
 
 # configuration of the Model Control Plane
-model_config:
+model_version:
   name: e2e_use_case
   version: staging
 

diff --git a/examples/e2e/configs/inference_config.yaml b/examples/e2e/configs/inference_config.yaml
@@ -34,7 +34,7 @@ steps:
       notify_on_success: False
 
 # configuration of the Model Control Plane
-model_config:
+model_version:
   name: e2e_use_case
   version: staging
 

diff --git a/examples/e2e/configs/train_config.yaml b/examples/e2e/configs/train_config.yaml
@@ -44,7 +44,7 @@ steps:
       notify_on_success: False
 
 # configuration of the Model Control Plane
-model_config:
+model_version:
   name: e2e_use_case
   license: apache
   description: e2e_use_case E2E Batch Use Case
@@ -61,7 +61,6 @@ model_config:
   - sklearn
   - from template
   - ZenML delivered
-  create_new_model_version: true
 
 # pipeline level extra configurations
 extra:

diff --git a/examples/e2e/inference_config.yaml b/examples/e2e/inference_config.yaml
diff --git a/examples/e2e/pipelines/training.py b/examples/e2e/pipelines/training.py
@@ -101,7 +101,7 @@ def e2e_use_case_training(
             target=target,
         )
         after.append(step_name)
-    best_model = hp_tuning_select_best_model(after=after)
+    best_model = hp_tuning_select_best_model(step_names=after, after=after)
 
     ########## Training stage ##########
     model = model_trainer(