diff --git a/notebooks/official/ml_metadata/vertex-pipelines-ml-metadata.ipynb b/notebooks/official/ml_metadata/vertex-pipelines-ml-metadata.ipynb index 9fa2d8701..a01c2e8be 100644 --- a/notebooks/official/ml_metadata/vertex-pipelines-ml-metadata.ipynb +++ b/notebooks/official/ml_metadata/vertex-pipelines-ml-metadata.ipynb @@ -719,7 +719,7 @@ "outputs": [], "source": [ "@component(\n", - " packages_to_install=[\"sklearn\", \"pandas\", \"joblib\"],\n", + " packages_to_install=[\"scikit-learn\", \"pandas\", \"joblib\"],\n", " base_image=\"python:3.9\",\n", " output_component_file=\"beans_model_component.yaml\",\n", ")\n",