This release updates Framework model regex for ModelPackage to suppor…

…t new Framework version xgboost, sklearn.
aws · Nov 2, 2022 · 2cedf2b · 2cedf2b
1 parent 4b76539
commit 2cedf2b
Show file tree

Hide file tree

Showing 6 changed files with 260 additions and 243 deletions.
diff --git a/generator/ServiceModels/sagemaker/sagemaker-2017-07-24.api.json b/generator/ServiceModels/sagemaker/sagemaker-2017-07-24.api.json
@@ -12866,7 +12866,7 @@
         "Environment":{"shape":"EnvironmentMap"},
         "ModelInput":{"shape":"ModelInput"},
         "Framework":{"shape":"String"},
-        "FrameworkVersion":{"shape":"FrameworkVersion"},
+        "FrameworkVersion":{"shape":"ModelPackageFrameworkVersion"},
         "NearestModelName":{"shape":"String"}
       }
     },
@@ -12876,6 +12876,12 @@
       "max":15,
       "min":1
     },
+    "ModelPackageFrameworkVersion":{
+      "type":"string",
+      "max":10,
+      "min":3,
+      "pattern":"[0-9]\\.[A-Za-z0-9.-]+"
+    },
     "ModelPackageGroup":{
       "type":"structure",
       "members":{

diff --git a/generator/ServiceModels/sagemaker/sagemaker-2017-07-24.docs.json b/generator/ServiceModels/sagemaker/sagemaker-2017-07-24.docs.json
@@ -5095,8 +5095,7 @@
     "FrameworkVersion": {
       "base": null,
       "refs": {
-        "InputConfig$FrameworkVersion": "<p>Specifies the framework version to use. This API field is only supported for the MXNet, PyTorch, TensorFlow and TensorFlow Lite frameworks.</p> <p>For information about framework versions supported for cloud targets and edge devices, see <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/neo-supported-cloud.html\">Cloud Supported Instance Types and Frameworks</a> and <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/neo-supported-devices-edge-frameworks.html\">Edge Supported Frameworks</a>.</p>",
-        "ModelPackageContainerDefinition$FrameworkVersion": "<p>The framework version of the Model Package Container Image.</p>"
+        "InputConfig$FrameworkVersion": "<p>Specifies the framework version to use. This API field is only supported for the MXNet, PyTorch, TensorFlow and TensorFlow Lite frameworks.</p> <p>For information about framework versions supported for cloud targets and edge devices, see <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/neo-supported-cloud.html\">Cloud Supported Instance Types and Frameworks</a> and <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/neo-supported-devices-edge-frameworks.html\">Edge Supported Frameworks</a>.</p>"
       }
     },
     "GenerateCandidateDefinitionsOnly": {
@@ -7609,6 +7608,12 @@
         "InferenceSpecification$Containers": "<p>The Amazon ECR registry path of the Docker image that contains the inference code.</p>"
       }
     },
+    "ModelPackageFrameworkVersion": {
+      "base": null,
+      "refs": {
+        "ModelPackageContainerDefinition$FrameworkVersion": "<p>The framework version of the Model Package Container Image.</p>"
+      }
+    },
     "ModelPackageGroup": {
       "base": "<p>A group of versioned models in the model registry.</p>",
       "refs": {