From d4d52f918e7a1b34a8113824d5d3c6a5ef7e251a Mon Sep 17 00:00:00 2001
From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com>
Date: Wed, 11 Oct 2023 18:40:32 +0200
Subject: [PATCH] Reduce integrations to local (#13)

* reduce integrations to local

* rework test action

* clean-up after run

* fix extra typo
---
 .github/actions/e2e_template_test/action.yml | 13 +++++++++----
 .github/workflows/ci.yml                     |  2 +-
 template/Makefile                            |  2 +-
 3 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/.github/actions/e2e_template_test/action.yml b/.github/actions/e2e_template_test/action.yml
index 6780cb6..9802e08 100644
--- 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/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 3dbe72b..abb636c 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-2300-model-watch-tower-v0.1
+          ref-zenml: develop
diff --git a/template/Makefile b/template/Makefile
index 3f0ab4a..a5672fe 100644
--- 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)" && \