refactor CI/CD

ivy-llc · Oct 25, 2023 · 0de04bd · 0de04bd
1 parent 2756ccf
commit 0de04bd
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/_basic-test.yml b/.github/workflows/_basic-test.yml
@@ -2,7 +2,7 @@ name: test-basics
 on: [workflow_call, workflow_dispatch]
 
 jobs:
-  Ubuntu:
+  cpu:
     runs-on: ubuntu-latest #ToDo: ${{ needs.build.outputs.os }}
     strategy:
       fail-fast: false

diff --git a/.github/workflows/_demo-test.yml b/.github/workflows/_demo-test.yml
@@ -2,7 +2,7 @@ name: test-examples
 on: [workflow_call, workflow_dispatch]
 
 jobs:
-  GCP:
+  gpu:
     runs-on: self-hosted #ToDo: ${{ needs.build.outputs.os }}
     strategy:
       fail-fast: false

diff --git a/tests/notebook_testing.py b/tests/notebook_testing.py
@@ -170,8 +170,8 @@ def startTest(self, test):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("-notebook_path", help="Path to the notebook file")
-    parser.add_argument("-module", help="Can either test examples or Basics")
+    parser.add_argument("notebook_path", help="Path to the notebook file")
+    parser.add_argument("module", help="Can either test examples or Basics")
     args = parser.parse_args()
 
     NotebookTest.notebook = args.notebook_path

diff --git a/tests/test_demos.sh b/tests/test_demos.sh
@@ -14,4 +14,5 @@ echo -n "$1" > .ivy/key.pem
 python3 -m pip install -r requirements.txt >/dev/null 2>&1
 
 # run test
-python3 tests/notebook_testing.py -notebook-path "$2" -module "$3"
+echo "$2" "$3"
+python3 tests/notebook_testing.py "$2" "$3"