Merge branch 'main' into feature/OSS-2494-yaml-config

zenml-io · Oct 11, 2023 · 9904bee · 9904bee
2 parents 074f0bc + d4d52f9
commit 9904bee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/.github/actions/e2e_template_test/action.yml b/.github/actions/e2e_template_test/action.yml
@@ -58,19 +58,19 @@ runs:
       if: ${{ inputs.ref-zenml != '' }}
       shell: bash
       run: |
-        pip install "git+https://github.com/zenml-io/zenml.git@${{ inputs.ref-zenml }}" "zenml[server]@git+https://github.com/zenml-io/zenml.git@${{ inputs.ref-zenml }}"
+        pip install "zenml[dev, server, templates]@git+https://github.com/zenml-io/zenml.git@${{ inputs.ref-zenml }}"
     
     - name: Install ZenML
       if: ${{ inputs.ref-zenml == '' }}
       shell: bash
       run: |
-        pip install zenml "zenml[server]"
+        pip install "zenml[dev, server, templates]"
     
     - name: Concatenate requirements
       shell: bash
       run: |
-        zenml integration export-requirements -o ./local_checkout/integration-requirements.txt sklearn mlflow s3 kubernetes kubeflow slack evidently
-        cat ./local_checkout/requirements.txt ./local_checkout/test-requirements.txt ./local_checkout/integration-requirements.txt >> ./local_checkout/all-requirements.txt
+        zenml integration export-requirements -o ./local_checkout/integration-requirements.txt sklearn mlflow s3 slack evidently
+        cat ./local_checkout/test-requirements.txt ./local_checkout/integration-requirements.txt >> ./local_checkout/all-requirements.txt
 
     - name: Install requirements
       shell: bash
@@ -83,3 +83,8 @@ runs:
         ZENML_STACK_NAME: ${{ inputs.stack-name }}
       run: |
         pytest ./local_checkout/tests
+
+    - name: Clean-up
+      shell: bash
+      run: |
+        rm -rf ./local_checkout
diff --git a/template/Makefile b/template/Makefile
@@ -9,7 +9,7 @@ setup: remote-login
 setup:
 {%- endif %}
 	pip install -r requirements.txt
-	zenml integration install sklearn mlflow s3 kubernetes kubeflow slack evidently -y
+	zenml integration install sklearn mlflow slack evidently -y
 
 install-stack:
 	@echo "Specify stack name [$(stack_name)]: " && read input && [ -n "$$input" ] && stack_name="$$input" || stack_name="$(stack_name)" && \