updated to 4.2

aws · Jun 9, 2022 · ea12b66 · ea12b66
1 parent 939a09f
commit ea12b66
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/advanced_functionality/autogluon-sagemaker-pipeline/pipelines/abalone/pipeline.py b/advanced_functionality/autogluon-sagemaker-pipeline/pipelines/abalone/pipeline.py
@@ -46,10 +46,10 @@
 
 # official autogluon images
 image_uri = (
-    "763104351884.dkr.ecr.us-east-1.amazonaws.com/autogluon-training:0.3.1-cpu-py37-ubuntu18.04"
+    "763104351884.dkr.ecr.us-east-1.amazonaws.com/autogluon-training:0.4.2-cpu-py38-ubuntu20.04"
 )
 infere_image_uri = (
-    "763104351884.dkr.ecr.us-east-1.amazonaws.com/autogluon-inference:0.3.1-cpu-py37-ubuntu16.04"
+    "763104351884.dkr.ecr.us-east-1.amazonaws.com/autogluon-inference:0.4.2-cpu-py38-ubuntu20.04"
 )
 
 
@@ -175,8 +175,8 @@ def get_pipeline(
         region=region,
         instance_count=1,
         instance_type=training_instance_type,
-        framework_version="0.3.1",
-        py_version="py37",
+        framework_version="0.4",
+        py_version="py38",
         base_job_name=f"{base_job_prefix}/abalone-train",
         output_path=model_path,
     )

diff --git a/advanced_functionality/autogluon-sagemaker-pipeline/pipelines/ag_model.py b/advanced_functionality/autogluon-sagemaker-pipeline/pipelines/ag_model.py
@@ -71,13 +71,13 @@ def __init__(self, *args, **kwargs):
 
 class AutoGluonInferenceModel(MXNetModel):
     def __init__(
-        self, model_data, role, entry_point, region, framework_version, instance_type, **kwargs
+        self, model_data, role, entry_point, region, framework_version, py_version, instance_type, **kwargs,
     ):
         image_uri = image_uris.retrieve(
             "autogluon",
             region=region,
             version=framework_version,
-            py_version="py37",
+            py_version=py_version,
             image_scope="inference",
             instance_type=instance_type,
         )