test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Apr 23, 2024 · 82a3587 · 82a3587
1 parent e3697d6
commit 82a3587
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 9 deletions.
diff --git a/.github/workflows/checks_new.yml b/.github/workflows/checks_new.yml
@@ -26,7 +26,7 @@ jobs:
         run: |
           echo "PACKAGES=$(find examples -maxdepth 1 -type d -exec basename '{}' \; \
           | grep -v -e 'testing' -e 'examples' \
-          | grep -v -e 'airflow_plugin' -e 'forecasting_sales' \
+          | grep -v -e 'airflow_plugin' -e 'forecasting_sales' -e 'onnx_plugin' \
           | sort \
           | jq --raw-input . \
           | jq --slurp . \
@@ -58,21 +58,22 @@ jobs:
           # Look to see if there is a cache hit for the corresponding requirements files
           key: ${{ format('{0}-pip-{1}', runner.os, hashFiles('dev-requirements.in', 'requirements.in')) }}
       - name: Install dependencies
+        working-directory: ${{ matrix.example }}
         run: |
-          sudo apt-get update && sudo apt install python3-packaging
           pip install uv
           uv venv
           source .venv/bin/activate
-          uv pip install -r requirements.in
+          uv pip install flytekit flytekitplugins-envd
+          if [ -f requirements.in ]; then uv pip install -r requirements.in; fi
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v1
         with:
           registry: ghcr.io
           username: ${{ secrets.FLYTE_BOT_USERNAME }}
           password: ${{ secrets.FLYTE_BOT_PAT }}
       - name: Pyflyte package
+        working-directory: ${{ matrix.example }}
         run: |
-          pip list
+          pip freeze
           source .venv/bin/activate
-          cd examples/${{ matrix.example }}
           pyflyte -vvv --pkgs ${{ matrix.example }} package -o spark-package.tgz --force --fast
diff --git a/examples/mmcloud_agent/requirements.in b/examples/mmcloud_agent/requirements.in
@@ -1,2 +1 @@
-flytekitplugins-envd
 flytekitplugins-mmcloud
diff --git a/examples/mmcloud_plugin/requirements.in b/examples/mmcloud_plugin/requirements.in
@@ -1,2 +1 @@
-flytekitplugins-envd
 flytekitplugins-mmcloud
diff --git a/examples/ray_plugin/requirements.in b/examples/ray_plugin/requirements.in
@@ -1,2 +1 @@
 flytekitplugins-ray
-flytekitplugins-envd
diff --git a/examples/snowflake_plugin/requirements.in b/examples/snowflake_plugin/requirements.in
@@ -1 +1,2 @@
-flytekitplugins-snowflake
+flytekitplugins-snowflake==1.7.0
+flytekit==1.7.0