sagemaker-core 1.0.15__py3-none-any.whl → 1.0.17__py3-none-any.whl

sagemaker_core/tools/resources_codegen.py

@@ -931,7 +931,9 @@ class ResourcesCodeGen:
             for e in errors:
                 error_shape = e["shape"]
                 error_shape_dict = self.shapes[error_shape]
-                error_shape_documentation = error_shape_dict.get("documentation").strip()
+                error_shape_documentation = error_shape_dict.get("documentation")
+                if error_shape_documentation:
+                    error_shape_documentation.strip()
                 shape_errors_and_docstrings[error_shape] = error_shape_documentation
         sorted_keys = sorted(shape_errors_and_docstrings.keys())
         return {key: shape_errors_and_docstrings[key] for key in sorted_keys}
@@ -939,7 +941,10 @@ class ResourcesCodeGen:
     def _exception_docstring(self, operation: str) -> str:
         _docstring = RESOURCE_METHOD_EXCEPTION_DOCSTRING
         for error, documentaion in self._fetch_shape_errors_and_doc_strings(operation).items():
-            _docstring += f"\n    {error}: {remove_html_tags(documentaion).strip()}"
+            if documentaion:
+                _docstring += f"\n    {error}: {remove_html_tags(documentaion).strip()}"
+            else:
+                _docstring += f"\n    {error}"
         return _docstring
 
     def _generate_docstring(

sagemaker_core/tools/shapes_extractor.py

@@ -119,19 +119,24 @@ class ShapesExtractor:
 
     def _evaluate_list_type(self, member_shape):
         list_shape_name = member_shape["member"]["shape"]
-        list_shape_type = self.combined_shapes[list_shape_name]["type"]
-        if list_shape_type in ["list", "map"]:
-            raise Exception(
-                "Unhandled list shape key type encountered, needs extra logic to handle this"
-            )
-        if list_shape_type == "structure":
+        list_shape_member = self.combined_shapes[list_shape_name]
+        list_shape_type = list_shape_member["type"]
+        if list_shape_type == "list":
+            member_type = f"List[{self._evaluate_list_type(list_shape_member)}]"
+        elif list_shape_type == "map":
+            member_type = f"List[{self._evaluate_map_type(list_shape_member)}]"
+        elif list_shape_type == "structure":
             # handling an edge case of nested structure
             if list_shape_name == "SearchExpression":
                 member_type = f"List['{list_shape_name}']"
             else:
                 member_type = f"List[{list_shape_name}]"
-        else:
+        elif list_shape_type in BASIC_JSON_TYPES_TO_PYTHON_TYPES.keys():
             member_type = f"List[{BASIC_JSON_TYPES_TO_PYTHON_TYPES[list_shape_type]}]"
+        else:
+            raise Exception(
+                f"Unhandled list shape key type {list_shape_type} for Shape: {list_shape_name} encountered, needs extra logic to handle this"
+            )
         return member_type
 
     def _evaluate_map_type(self, member_shape):

{sagemaker_core-1.0.15.dist-info → sagemaker_core-1.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.15
+Version: 1.0.17
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git
@@ -15,19 +15,19 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.8
 Description-Content-Type: text/x-rst
 License-File: LICENSE
-Requires-Dist: boto3 <2.0.0,>=1.34.0
-Requires-Dist: pydantic <3.0.0,>=2.0.0
-Requires-Dist: PyYAML <7.0,>=6.0
-Requires-Dist: jsonschema <5.0.0
-Requires-Dist: platformdirs <5.0.0,>=4.0.0
-Requires-Dist: rich <14.0.0,>=13.0.0
-Requires-Dist: mock <5.0,>4.0
-Requires-Dist: importlib-metadata <7.0,>=1.4.0
+Requires-Dist: boto3<2.0.0,>=1.35.75
+Requires-Dist: pydantic<3.0.0,>=2.0.0
+Requires-Dist: PyYAML<7.0,>=6.0
+Requires-Dist: jsonschema<5.0.0
+Requires-Dist: platformdirs<5.0.0,>=4.0.0
+Requires-Dist: rich<14.0.0,>=13.0.0
+Requires-Dist: mock<5.0,>4.0
+Requires-Dist: importlib-metadata<7.0,>=1.4.0
 Provides-Extra: codegen
-Requires-Dist: black <25.0.0,>=24.3.0 ; extra == 'codegen'
-Requires-Dist: pandas <3.0.0,>=2.0.0 ; extra == 'codegen'
-Requires-Dist: pytest <9.0.0,>=8.0.0 ; extra == 'codegen'
-Requires-Dist: pylint <4.0.0,>=3.0.0 ; extra == 'codegen'
+Requires-Dist: black<25.0.0,>=24.3.0; extra == "codegen"
+Requires-Dist: pandas<3.0.0,>=2.0.0; extra == "codegen"
+Requires-Dist: pytest<9.0.0,>=8.0.0; extra == "codegen"
+Requires-Dist: pylint<4.0.0,>=3.0.0; extra == "codegen"
 
 .. image:: https://github.com/aws/sagemaker-python-sdk/raw/master/branding/icon/sagemaker-banner.png
     :height: 100px

{sagemaker_core-1.0.15.dist-info → sagemaker_core-1.0.17.dist-info}/RECORD

@@ -3,33 +3,33 @@ sagemaker_core/_version.py,sha256=t-XEKT8_pvKkl5z_UrfcIU256c_NxaI16qGWMuNVpbM,32
 sagemaker_core/helper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker_core/helper/session_helper.py,sha256=GO1UJgpN1L9a25nYlVb-KWk4KvmFzVkLqFMqw-VaI4c,33126
 sagemaker_core/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sagemaker_core/main/config_schema.py,sha256=TeGoTobT4yotEVyfguLF0IdKYlOymsDZ45ySxXiCDuw,56998
+sagemaker_core/main/config_schema.py,sha256=kfQvf_japrsitTwXTdUaYhOPqKTBBwIE_-ebNeWD-yY,57229
 sagemaker_core/main/exceptions.py,sha256=87DUlrmHxaWoiYNlpNY9ixxFMPRk_dIGPsA2e_xdVwQ,5602
 sagemaker_core/main/intelligent_defaults_helper.py,sha256=5SDM6UavZtp-k5LhqRL7GRIDgzFB5UsC_p7YuiSPK9A,8334
 sagemaker_core/main/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
-sagemaker_core/main/resources.py,sha256=idwGeRbSSO7Acwtd0y_zOkIxOrMlJvwxx-gL0k7GlS8,1323921
-sagemaker_core/main/shapes.py,sha256=Vge1sJb41xqpOFdCnzl8JNdIOwPflgr0Ioy2IKhP01g,699286
+sagemaker_core/main/resources.py,sha256=9SXDc6rvZKgO3E7Sx00YFaE9GSprLM8-7OPZI59R0wE,1404781
+sagemaker_core/main/shapes.py,sha256=BFyvKO-l6BVhlAr683fY_73s4Ed6oX7WOiiPVWKsiKs,722024
 sagemaker_core/main/user_agent.py,sha256=4sZybDXkzRoZnOnVDQ8p8zFTfiRJdsH7amDWInVQ4xU,2708
 sagemaker_core/main/utils.py,sha256=LCFDM6oxf6_e1i-_Dgtkm3ehl7YfoEpJ2kTTFTL6iOU,18471
 sagemaker_core/main/code_injection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker_core/main/code_injection/base.py,sha256=11_Jif0nOzfbLGlXaacKf-wcizzfS64U0OSZGoVffFU,1733
-sagemaker_core/main/code_injection/codec.py,sha256=2DjmeD2uND307UqDefvVEpE0rZ8yfFU3Bi3TvQCQveI,7658
+sagemaker_core/main/code_injection/codec.py,sha256=nA51E9iNWHyKou1G23rKSRL4WitdkFRbMuFkyrGHzKU,8428
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=Akn-5sH0T8ujXuiXEOJtq5agJuRk-tKoFP04qwFx-NM,660663
+sagemaker_core/main/code_injection/shape_dag.py,sha256=8NmYwxcnkAjiTK1oeOrmNMkhs9ZqBwWbxBCFs29ooxY,693934
 sagemaker_core/resources/__init__.py,sha256=EAYTFMN-nPjnPjjBbhIUeaL67FLKNPd7qbcbl9VIrws,31
 sagemaker_core/shapes/__init__.py,sha256=RnbIu9eTxKt-DNsOFJabrWIgrrtS9_SdAozP9JBl_ic,28
 sagemaker_core/tools/__init__.py,sha256=xX79JImxCVzrWMnjgntLCve2G5I-R4pRar5s20kT9Rs,56
 sagemaker_core/tools/codegen.py,sha256=mKWVi2pWnPxyIoWUEPYjEc9Gw7D9bCOrHqa00yzIZ1o,2005
 sagemaker_core/tools/constants.py,sha256=a2WjUDK7gzxgilZs99vp30qh4kQ-y6JKhrwwqVAA12o,3385
 sagemaker_core/tools/data_extractor.py,sha256=pNfmTA0NUA96IgfLrla7a36Qjc1NljbwgZYaOhouKqQ,2113
-sagemaker_core/tools/method.py,sha256=4Hmu4UWpiBgUTZljYdW1KIKDduDxf_nfhCyuWgLVMWI,717
-sagemaker_core/tools/resources_codegen.py,sha256=ASirF9UMkGAOYZrrZxmFqK3gYQ4YbVYdAiUcWt6qdII,84360
+sagemaker_core/tools/method.py,sha256=Ud2YeH2SPkj7xtIxBuUdRfQwCmovMUzGGkcIvzhpQeQ,805
+sagemaker_core/tools/resources_codegen.py,sha256=bnufJgmVUcIGyMifZKbtwaO_bG_0bN2_A7CYq629pNc,84550
 sagemaker_core/tools/resources_extractor.py,sha256=hN61ehZbPnhFW-2FIVDi7NsEz4rLvGr-WoglHQGfrug,14523
 sagemaker_core/tools/shapes_codegen.py,sha256=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
-sagemaker_core/tools/shapes_extractor.py,sha256=4KjgDmhlPM4G1f1NeYbORKlXs1s7Q_sm_NK31S_ROQ0,11950
+sagemaker_core/tools/shapes_extractor.py,sha256=GFy55JHGW0V8cwN5SL5gGLUxihLGswueyh5iNCn1sYk,12310
 sagemaker_core/tools/templates.py,sha256=yX2RQKeClgYwKS5Qu_mDpnWJIBCuj0yELrdm95aiTpk,23262
-sagemaker_core-1.0.15.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.15.dist-info/METADATA,sha256=dRiT-vcjQOwj8BMHrbEY-fgpLKAg34XWdiHPbqZynCc,4878
-sagemaker_core-1.0.15.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-sagemaker_core-1.0.15.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.15.dist-info/RECORD,,
+sagemaker_core-1.0.17.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.17.dist-info/METADATA,sha256=7ZbFj4VOtErNDYpF24UPtnMZ2cO1Gd768ZvoD0EJ0Sc,4863
+sagemaker_core-1.0.17.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+sagemaker_core-1.0.17.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.17.dist-info/RECORD,,

{sagemaker_core-1.0.15.dist-info → sagemaker_core-1.0.17.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.5.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any