Use explicit zenml stack set to be robust to existing stacks

zenml-io · Feb 2, 2023 · 0e651d9 · 0e651d9
1 parent 76800ed
commit 0e651d9
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 6 deletions.
diff --git a/2-1_Experiment_Tracking.ipynb b/2-1_Experiment_Tracking.ipynb
@@ -117,7 +117,10 @@
     "!zenml experiment-tracker register mlflow_tracker --flavor=mlflow\n",
     "\n",
     "# Create a new stack with MLflow experiment tracker in it\n",
-    "!zenml stack register mlflow_stack -a default -o default -e mlflow_tracker --set\n"
+    "!zenml stack register mlflow_stack -a default -o default -e mlflow_tracker\n",
+    "\n",
+    "# Set the new stack as active\n",
+    "!zenml stack set mlflow_stack\n"
    ]
   },
   {
@@ -317,7 +320,10 @@
     "!zenml experiment-tracker register wandb_tracker --flavor=wandb --api_key={WANDB_API_KEY} --entity={WANDB_ENTITY} --project_name={WANDB_PROJECT}\n",
     "\n",
     "# Create a new stack with W&B experiment tracker in it\n",
-    "!zenml stack register wandb_stack -a default -o default -e wandb_tracker --set"
+    "!zenml stack register wandb_stack -a default -o default -e wandb_tracker\n",
+    "\n",
+    "# Set the new stack as active\n",
+    "!zenml stack set wandb_stack"
    ]
   },
   {
@@ -429,7 +435,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.0"
+   "version": "3.10.0 (default, Nov 29 2022, 17:00:25) [Clang 14.0.0 (clang-1400.0.29.202)]"
   },
   "vscode": {
    "interpreter": {

diff --git a/2-2_Local_Deployment.ipynb b/2-2_Local_Deployment.ipynb
@@ -139,7 +139,10 @@
     "!zenml model-deployer register mlflow --flavor=mlflow\n",
     "\n",
     "# Create a new stack with MLflow components\n",
-    "!zenml stack register mlflow_stack -a default -o default -d mlflow -e mlflow_tracker --set"
+    "!zenml stack register mlflow_stack -a default -o default -d mlflow -e mlflow_tracker\n",
+    "\n",
+    "# Set the new stack as active\n",
+    "!zenml stack set mlflow_stack"
    ]
   },
   {

diff --git a/2-3_Inference_Pipelines.ipynb b/2-3_Inference_Pipelines.ipynb
@@ -44,7 +44,8 @@
     "!zenml init\n",
     "!zenml experiment-tracker register mlflow_tracker --flavor=mlflow\n",
     "!zenml model-deployer register mlflow --flavor=mlflow\n",
-    "!zenml stack register mlflow_stack -a default -o default -d mlflow -e mlflow_tracker --set\n",
+    "!zenml stack register mlflow_stack -a default -o default -d mlflow -e mlflow_tracker\n",
+    "!zenml stack set mlflow_stack\n",
     "\n",
     "%pip install pyparsing==2.4.2  # required for Colab\n",
     "\n",

diff --git a/3-1_Data_Skew.ipynb b/3-1_Data_Skew.ipynb
@@ -186,7 +186,8 @@
    "outputs": [],
    "source": [
     "!zenml data-validator register evidently_validator --flavor=evidently\n",
-    "!zenml stack register evidently_stack -a default -o default -dv evidently_validator --set"
+    "!zenml stack register evidently_stack -a default -o default -dv evidently_validator\n",
+    "!zenml stack set evidently_stack"
    ]
   },
   {