add branch support

outerbounds · May 19, 2024 · 8f32934 · 8f32934
1 parent 2a0f5b7
commit 8f32934
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -27,10 +27,10 @@ jobs:
           --perimeter default \
           --github-actions
 
-      - name: Run helloworld flow
+      - name: Run tests
         run: |-
           python hello.py run --with kubernetes
 
       - name: Deploy to production
         run: |-
-          python hello.py argo-workflows create
+          python hello.py --branch ${{ github.head_ref || github.ref_name }} argo-workflows create
diff --git a/hello.py b/hello.py
@@ -1,5 +1,6 @@
-from metaflow import FlowSpec, step, card, schedule
+from metaflow import FlowSpec, step, card, project
 
+@project(name='mlproject')
 class GitHubActionsDemo(FlowSpec):
 
     @step