Skip to content

fix: update with instructions to create mlflow app #5556

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kmanuwai wants to merge 1 commit into
base: master
Choose a base branch
from
Open

fix: update with instructions to create mlflow app #5556

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 10 additions & 11 deletions v3-examples/ml-ops-examples/v3-mlflow-train-inference-e2e-example.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,9 @@
"metadata": {},
"source": [
"### Prerequisites\n",
"- SageMaker MLflow App created (tracking server ARN required)\n",
"- IAM permissions for MLflow tracking and model registry\n",
"- SageMaker MLFlow App created (tracking server ARN required). \n",
" - Follow steps here to create a serverless MLflow app managed by SageMaker: https://docs.aws.amazon.com/sagemaker/latest/dg/mlflow-app-setup.html\n",
"- IAM permissions for MLFlow tracking and model registry\n",
"- AWS credentials configured"
]
},
Expand All @@ -25,7 +26,7 @@
"source": [
"## Step 0: Install Dependencies\n",
"\n",
"**Note:** There are known issues with MLflow model path resolution. Install the latest published SDK from GitHub for the latest fixes."
"**Note:** Version 3.4.0+ of the SageMaker SDK is required. "
]
},
{
Expand All @@ -34,24 +35,22 @@
"metadata": {},
"outputs": [],
"source": [
"# Install fix for MLflow path resolution issues\n",
"%pip install mlflow==3.4.0"
"# Install SageMaker, MLFlow & the SageMaker MLFlow Plugin \n",
"%pip install \"sagemaker>=3.4.0\" mlflow==3.4.0 sagemaker-mlflow==0.2.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"#### NOTE: You must restart your kernel"
"##### NOTE: restart your kernel"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Step 1: Configuration\n",
"\n",
"Set up MLflow tracking server and training configuration."
"## Step 1: Configuration\n"
]
},
{
Expand Down Expand Up @@ -557,7 +556,7 @@
],
"metadata": {
"kernelspec": {
"display_name": "py3.10.14",
"display_name": ".venv",
"language": "python",
"name": "python3"
},
Expand All @@ -571,7 +570,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.14"
"version": "3.11.14"
}
},
"nbformat": 4,
Expand Down