Merge pull request #27 from felipenv/main

ML-9956 removing default image

Author: daniels290813

financial-payment-pipeline.ipynb

@@ -8,6 +8,19 @@
     "# Payment classification pipeline"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a3f980d2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install --upgrade mlrun --quiet # upgrade mlrun to the latest vesion\n",
+    "%pip install --upgrade sagemaker --quiet # upgrade SageMaker to the latest vesion\n",
+    "%pip install \"kfp==1.8.22\" \"kfp-server-api==1.8.5\" --quiet\n",
+    "%pip install \"protobuf<4,>=3.20.0\" --force-reinstall --quiet"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
@@ -26,7 +39,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "id": "e34f8c80-6584-4e80-981c-0f17e1584ebf",
    "metadata": {
     "tags": []
@@ -46,7 +59,6 @@
     "    user_project=True,\n",
     "    parameters={\n",
     "        # \"source\" : \"git://github.com/mlrun/demo-sagemaker#main\",\n",
-    "        \"default_image\" : \"yonishelach/sagemaker-demo\"\n",
     "    }\n",
     ")"
    ]

project_setup.py

@@ -31,14 +31,8 @@ def setup(
 
     # Unpack parameters:
     source = project.get_param(key="source")
-    default_image = project.get_param(key="default_image")
 
-    # Set or build the default image:
-    if default_image is None and not project.spec.build.image:
-        print("Building default image for the demo:")
-        _build_image(project=project)
-    else:
-        project.set_default_image(default_image)
+    _build_image(project=project)
 
     # Set the project git source:
     if source:
@@ -71,6 +65,7 @@ def _build_image(project: mlrun.projects.MlrunProject):
             "pip install -U xgboost",
             "pip install -U tarsafe",
             "pip install -U aiobotocore",
+            "pip install -U mlrun[complete]",
         ],
         set_as_default=True,
     )

src/functions/train.py

@@ -38,7 +38,7 @@ def train(context):
         container,
         role,
         instance_count=1,
-        instance_type="ml.m4.xlarge",
+        instance_type="ml.m5.xlarge",
         output_path="s3://{}/{}/output".format(s3_bucket, bucket_prefix),
         sagemaker_session=sagemaker_session,
     )