Add all example notebooks to the run_notebooks Github Action (#577)

RunLLM · Oct 17, 2022 · e13f5b2 · e13f5b2
1 parent 8665e33
commit e13f5b2
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 6 deletions.
diff --git a/.github/workflows/run-notebooks.yml b/.github/workflows/run-notebooks.yml
@@ -79,15 +79,40 @@ jobs:
         timeout-minutes: 10
         run: python3 run_notebook.py --path "tutorials/Parameters Tutorial.ipynb"
 
-      - name: Run Sentiment Notebook
+      - name: Run Churn Tutorial Notebook
+        working-directory: examples
+        timeout-minutes: 10
+        run: python3 run_notebook.py --path "churn_prediction/Customer Churn Prediction.ipynb"
+
+      - name: Run Sentiment Analysis Notebook
         working-directory: examples
         timeout-minutes: 10
         run: python3 run_notebook.py --path "sentiment-analysis/Sentiment Model.ipynb"
 
-      - name: Run Churn Tutorial Notebook
+      # TODO(ENG-1821): Remove this once version 0.1 is released.
+      - name: Populate Housing and Diabetes Data
+        timeout-minutes: 1
+        run: python3 src/resources/demo_db/sqlite/v0.1/create_tables.py
+
+      - name: Run Classifying Diabetes Risk Notebook
         working-directory: examples
         timeout-minutes: 10
-        run: python3 run_notebook.py --path "churn_prediction/Customer Churn Prediction.ipynb"
+        run: python3 run_notebook.py --path "diabetes-classifier/Classifying Diabetes Risk.ipynb"
+
+      - name: Run House Price Prediction Notebook
+        working-directory: examples
+        timeout-minutes: 10
+        run: python3 run_notebook.py --path "house-price-prediction/House Price Prediction.ipynb"
+
+      - name: Run MPG Regressor Notebook
+        working-directory: examples
+        timeout-minutes: 10
+        run: python3 run_notebook.py --path "mpg-regressor/Predicting MPG.ipynb"
+
+      - name: Run Wine Ratings Prediction Notebook
+        working-directory: examples
+        timeout-minutes: 10
+        run: python3 run_notebook.py --path "wine-ratings-prediction/Predict Missing Wine Ratings.ipynb"
 
       - name: Run Imported Function Notebook
         timeout-minutes: 10

diff --git a/sdk/aqueduct/aqueduct_client.py b/sdk/aqueduct/aqueduct_client.py
@@ -296,11 +296,11 @@ def flow(self, flow_id: Union[str, uuid.UUID]) -> Flow:
     def publish_flow(
         self,
         name: str,
-        artifacts: Union[BaseArtifact, List[BaseArtifact]],
-        metrics: Optional[List[NumericArtifact]] = None,
-        checks: Optional[List[BoolArtifact]] = None,
         description: str = "",
         schedule: str = "",
+        artifacts: Optional[Union[BaseArtifact, List[BaseArtifact]]] = None,
+        metrics: Optional[List[NumericArtifact]] = None,
+        checks: Optional[List[BoolArtifact]] = None,
         config: Optional[FlowConfig] = None,
     ) -> Flow:
         """Uploads and kicks off the given flow in the system.