diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 9be8bfc..a14241e 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -55,5 +55,5 @@ jobs:
with:
stack-name: ${{ matrix.stack-name }}
python-version: ${{ matrix.python-version }}
- ref-zenml: ${{ inputs.ref-zenml || 'feature/PRD-566-dependency-cleanup' }}
+ ref-zenml: ${{ inputs.ref-zenml || 'develop' }}
ref-template: ${{ inputs.ref-template || github.ref }}
diff --git a/template/README.md b/template/README.md
index 10dc01e..b25258f 100644
--- a/template/README.md
+++ b/template/README.md
@@ -24,7 +24,7 @@ Along the way we will also show you how to:
You can use Google Colab to see ZenML in action, no signup / installation required!
-
+
## :computer: Run Locally
@@ -36,7 +36,7 @@ pip install "zenml[server]"
# clone the ZenML repository
git clone https://github.com/zenml-io/zenml.git
-cd zenml/examples/quickstart
+cd zenml/examples/mlops_starter
```
Now we're ready to start. You have two options for running the quickstart locally:
@@ -45,13 +45,13 @@ Now we're ready to start. You have two options for running the quickstart locall
```bash
pip install notebook
jupyter notebook
-# open notebooks/quickstart.ipynb
+# open quickstart.ipynb
```
#### Option 2 - Execute the whole ML pipeline from a Python script:
```bash
# Install required zenml integrations
-zenml integration install sklearn -y
+zenml integration install sklearn pandas -y
# Initialize ZenML
zenml init
diff --git a/template/configs/feature_engineering.yaml b/template/configs/feature_engineering.yaml
index d5ab212..055bb6f 100644
--- a/template/configs/feature_engineering.yaml
+++ b/template/configs/feature_engineering.yaml
@@ -3,6 +3,7 @@ settings:
docker:
required_integrations:
- sklearn
+ - pandas
requirements:
- pyarrow
diff --git a/template/configs/inference.yaml b/template/configs/inference.yaml
index 1dcefe4..8f73d76 100644
--- a/template/configs/inference.yaml
+++ b/template/configs/inference.yaml
@@ -3,6 +3,7 @@ settings:
docker:
required_integrations:
- sklearn
+ - pandas
requirements:
- pyarrow
diff --git a/template/configs/training_rf.yaml b/template/configs/training_rf.yaml
index 8d75610..70fa641 100644
--- a/template/configs/training_rf.yaml
+++ b/template/configs/training_rf.yaml
@@ -3,6 +3,7 @@ settings:
docker:
required_integrations:
- sklearn
+ - pandas
requirements:
- pyarrow
diff --git a/template/configs/training_sgd.yaml b/template/configs/training_sgd.yaml
index 857cdf7..386b53b 100644
--- a/template/configs/training_sgd.yaml
+++ b/template/configs/training_sgd.yaml
@@ -3,6 +3,7 @@ settings:
docker:
required_integrations:
- sklearn
+ - pandas
requirements:
- pyarrow
diff --git a/template/quickstart.ipynb b/template/quickstart.ipynb
index afcf2ca..f2f09be 100644
--- a/template/quickstart.ipynb
+++ b/template/quickstart.ipynb
@@ -31,7 +31,7 @@
"required!\n",
"\n",
"[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](\n",
- "https://colab.research.google.com/github/zenml-io/zenml/blob/main/examples/quickstart/quickstart.ipynb)"
+ "https://colab.research.google.com/github/zenml-io/zenml/blob/main/examples/mlops_starter/quickstart.ipynb)"
]
},
{
diff --git a/template/requirements.txt b/template/requirements.txt
index d38ead5..1e0a8ac 100644
--- a/template/requirements.txt
+++ b/template/requirements.txt
@@ -2,3 +2,4 @@ zenml[server]>=0.50.0
notebook
scikit-learn
pyarrow
+pandas