Optimize functional tests registration

Signed-off-by: eduardo apolinario <eapolinario@users.noreply.github.com>
flyteorg · Jul 25, 2023 · 91b743e · 91b743e
1 parent 66a9317
commit 91b743e
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/.github/workflows/end2end.yml b/.github/workflows/end2end.yml
@@ -50,13 +50,14 @@ jobs:
           pip install flytekit
           pip freeze
       - name: Checkout flytesnacks
+        if: ${{ inputs.priorities == 'P0' }}
         uses: actions/checkout@v3
         with:
           repository: flyteorg/flytesnacks
           path: flytesnacks
-          # TODO: Enable this once refactored version produces a release tag
-          # ref: ${{ env.FLYTESNACKS_VERSION }}
-      - name: Register specific tests
+          ref: ${{ env.FLYTESNACKS_VERSION }}
+      - name: Register P0 tests
+        if: ${{ inputs.priorities == 'P0' }}
         run: |
           for f in basics/basics/basic_workflow.py \
                   basics/basics/hello_world.py \
@@ -70,14 +71,22 @@ jobs:
                   type_system/type_system/schema.py \
                   type_system/type_system/typed_schema.py ;
           do
-              pyflyte --config ./boilerplate/flyte/end2end/functional-test-config.yaml \
-                  register \
-                  --project flytesnacks \
-                  --domain development \
-                  --image cr.flyte.org/flyteorg/flytekit:py3.11-latest \
-                  --version ${{ env.FLYTESNACKS_VERSION }} \
-                  flytesnacks/examples/$f;
+            pyflyte --config ./boilerplate/flyte/end2end/functional-test-config.yaml \
+                register \
+                --project flytesnacks \
+                --domain development \
+                --image cr.flyte.org/flyteorg/flytekit:py3.11-latest \
+                --version ${{ env.FLYTESNACKS_VERSION }} \
+                flytesnacks/examples/$f;
           done
+       - name: Register all flytesnacks examples
+         if: ${{ inputs.priorities != 'P0' }}
+         uses: unionai/flyte-register-action@v0.0.2
+         with:
+           flytesnacks: true
+           project: flytesnacks
+           version: ${{ env.FLYTESNACKS_VERSION }}
+           domain: development
       - name: Pre Upgrade Tests
         if: ${{ github.event.repository.name == 'flyteadmin' }}
         env: