sagemaker-core 1.0.7__py3-none-any.whl → 1.0.8__py3-none-any.whl

sagemaker_core/main/code_injection/shape_dag.py

@@ -6573,7 +6573,24 @@ SHAPE_DAG = {
     },
     "Groups": {"member_shape": "Group", "member_type": "string", "type": "list"},
     "HiddenAppTypesList": {"member_shape": "AppType", "member_type": "string", "type": "list"},
+    "HiddenInstanceTypesList": {
+        "member_shape": "AppInstanceType",
+        "member_type": "string",
+        "type": "list",
+    },
     "HiddenMlToolsList": {"member_shape": "MlTools", "member_type": "string", "type": "list"},
+    "HiddenSageMakerImage": {
+        "members": [
+            {"name": "SageMakerImageName", "shape": "SageMakerImageName", "type": "string"},
+            {"name": "VersionAliases", "shape": "VersionAliasesList", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "HiddenSageMakerImageVersionAliasesList": {
+        "member_shape": "HiddenSageMakerImage",
+        "member_type": "structure",
+        "type": "list",
+    },
     "HolidayConfig": {
         "member_shape": "HolidayConfigAttributes",
         "member_type": "structure",
@@ -13135,6 +13152,12 @@ SHAPE_DAG = {
         "members": [
             {"name": "HiddenMlTools", "shape": "HiddenMlToolsList", "type": "list"},
             {"name": "HiddenAppTypes", "shape": "HiddenAppTypesList", "type": "list"},
+            {"name": "HiddenInstanceTypes", "shape": "HiddenInstanceTypesList", "type": "list"},
+            {
+                "name": "HiddenSageMakerImageVersionAliases",
+                "shape": "HiddenSageMakerImageVersionAliasesList",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -14701,6 +14724,11 @@ SHAPE_DAG = {
         "members": [{"name": "Dimension", "shape": "Dimension", "type": "integer"}],
         "type": "structure",
     },
+    "VersionAliasesList": {
+        "member_shape": "ImageVersionAliasPattern",
+        "member_type": "string",
+        "type": "list",
+    },
     "Vertex": {
         "members": [
             {"name": "Arn", "shape": "AssociationEntityArn", "type": "string"},

sagemaker_core/main/resources.py

@@ -16518,6 +16518,7 @@ class Model(Base):
                         "s3_data_source": {
                             "s3_uri": {"type": "string"},
                             "s3_data_type": {"type": "string"},
+                            "manifest_s3_uri": {"type": "string"},
                         }
                     }
                 },
@@ -27953,6 +27954,55 @@ class TrialComponent(Base):
         response = client.batch_put_metrics(**operation_input_args)
         logger.debug(f"Response: {response}")
 
+    @classmethod
+    @Base.add_validate_call
+    def batch_get_metrics(
+        cls,
+        metric_queries: List[MetricQuery],
+        session: Optional[Session] = None,
+        region: Optional[str] = None,
+    ) -> Optional[BatchGetMetricsResponse]:
+        """
+        Used to retrieve training metrics from SageMaker.
+
+        Parameters:
+            metric_queries: Queries made to retrieve training metrics from SageMaker.
+            session: Boto3 session.
+            region: Region name.
+
+        Returns:
+            BatchGetMetricsResponse
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+        """
+
+        operation_input_args = {
+            "MetricQueries": metric_queries,
+        }
+        # serialize the input request
+        operation_input_args = serialize(operation_input_args)
+        logger.debug(f"Serialized input request: {operation_input_args}")
+
+        client = Base.get_sagemaker_client(
+            session=session, region_name=region, service_name="sagemaker-metrics"
+        )
+
+        logger.debug(f"Calling batch_get_metrics API")
+        response = client.batch_get_metrics(**operation_input_args)
+        logger.debug(f"Response: {response}")
+
+        transformed_response = transform(response, "BatchGetMetricsResponse")
+        return BatchGetMetricsResponse(**transformed_response)
+
 
 class UserProfile(Base):
     """

sagemaker_core/main/shapes.py

@@ -406,6 +406,18 @@ class MetricQueryResult(Base):
     message: Optional[str] = Unassigned()
 
 
+class BatchGetMetricsResponse(Base):
+    """
+    BatchGetMetricsResponse
+
+    Attributes
+    ----------------------
+    metric_query_results: The results of a query to retrieve training metrics from SageMaker.
+    """
+
+    metric_query_results: Optional[List[MetricQueryResult]] = Unassigned()
+
+
 class BatchPutMetricsError(Base):
     """
     BatchPutMetricsError
@@ -4304,6 +4316,21 @@ class CustomFileSystemConfig(Base):
     efs_file_system_config: Optional[EFSFileSystemConfig] = Unassigned()
 
 
+class HiddenSageMakerImage(Base):
+    """
+    HiddenSageMakerImage
+      The SageMaker images that are hidden from the Studio user interface. You must specify the SageMaker image name and version aliases.
+
+    Attributes
+    ----------------------
+    sage_maker_image_name:  The SageMaker image name that you are hiding from the Studio user interface.
+    version_aliases:  The version aliases you are hiding from the Studio user interface.
+    """
+
+    sage_maker_image_name: Optional[str] = Unassigned()
+    version_aliases: Optional[List[str]] = Unassigned()
+
+
 class StudioWebPortalSettings(Base):
     """
     StudioWebPortalSettings
@@ -4313,10 +4340,14 @@ class StudioWebPortalSettings(Base):
     ----------------------
     hidden_ml_tools: The machine learning tools that are hidden from the Studio left navigation pane.
     hidden_app_types: The Applications supported in Studio that are hidden from the Studio left navigation pane.
+    hidden_instance_types:  The instance types you are hiding from the Studio user interface.
+    hidden_sage_maker_image_version_aliases:  The version aliases you are hiding from the Studio user interface.
     """
 
     hidden_ml_tools: Optional[List[str]] = Unassigned()
     hidden_app_types: Optional[List[str]] = Unassigned()
+    hidden_instance_types: Optional[List[str]] = Unassigned()
+    hidden_sage_maker_image_version_aliases: Optional[List[HiddenSageMakerImage]] = Unassigned()
 
 
 class UserSettings(Base):
@@ -5849,15 +5880,15 @@ class HumanTaskConfig(Base):
 
     workteam_arn: str
     ui_config: UiConfig
-    pre_human_task_lambda_arn: str
     task_title: str
     task_description: str
     number_of_human_workers_per_data_object: int
     task_time_limit_in_seconds: int
-    annotation_consolidation_config: AnnotationConsolidationConfig
+    pre_human_task_lambda_arn: Optional[str] = Unassigned()
     task_keywords: Optional[List[str]] = Unassigned()
     task_availability_lifetime_in_seconds: Optional[int] = Unassigned()
     max_concurrent_task_count: Optional[int] = Unassigned()
+    annotation_consolidation_config: Optional[AnnotationConsolidationConfig] = Unassigned()
     public_workforce_task_price: Optional[PublicWorkforceTaskPrice] = Unassigned()
 
 
@@ -9979,7 +10010,7 @@ class LabelingJobSummary(Base):
     labeling_job_status: str
     label_counters: LabelCounters
     workteam_arn: str
-    pre_human_task_lambda_arn: str
+    pre_human_task_lambda_arn: Optional[str] = Unassigned()
     annotation_consolidation_lambda_arn: Optional[str] = Unassigned()
     failure_reason: Optional[str] = Unassigned()
     labeling_job_output: Optional[LabelingJobOutput] = Unassigned()

{sagemaker_core-1.0.7.dist-info → sagemaker_core-1.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.7
+Version: 1.0.8
 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

{sagemaker_core-1.0.7.dist-info → sagemaker_core-1.0.8.dist-info}/RECORD

@@ -6,15 +6,15 @@ sagemaker_core/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 sagemaker_core/main/config_schema.py,sha256=TeGoTobT4yotEVyfguLF0IdKYlOymsDZ45ySxXiCDuw,56998
 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/resources.py,sha256=S4wuM705rkah3cFO-kfWAXPgYq7SI3PG2Qh6ZHtfTfk,1314805
-sagemaker_core/main/shapes.py,sha256=75mSyS_bGPAIcFFwWD5b_Hc4TBNFgNKJGKzcNnY33ro,694903
+sagemaker_core/main/resources.py,sha256=VjjmTLzPcjjg3c8-1g9Fv88Gpwimd9KuNq02378B6cE,1316658
+sagemaker_core/main/shapes.py,sha256=Groaj8psouJwQpRljmZIDeRM6fyU6p2J3wgjzcvQ_5k,696205
 sagemaker_core/main/user_agent.py,sha256=4sZybDXkzRoZnOnVDQ8p8zFTfiRJdsH7amDWInVQ4xU,2708
 sagemaker_core/main/utils.py,sha256=lXkJyiCow5uf32l0EmkimB0RKVk2BS7OM2fYoLsOfD4,18346
 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/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=hyvmKI2IWBEt21fCQrpSmi4d1YzOpfiG5CGNwRpsziY,656883
+sagemaker_core/main/code_injection/shape_dag.py,sha256=0OoQzH_r_TAYR2-KUKGQnGMD4U9Cb-NRMxc_SORIOdw,657895
 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
@@ -27,8 +27,8 @@ sagemaker_core/tools/resources_extractor.py,sha256=hN61ehZbPnhFW-2FIVDi7NsEz4rLv
 sagemaker_core/tools/shapes_codegen.py,sha256=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
 sagemaker_core/tools/shapes_extractor.py,sha256=4KjgDmhlPM4G1f1NeYbORKlXs1s7Q_sm_NK31S_ROQ0,11950
 sagemaker_core/tools/templates.py,sha256=nze_A01EpegYUwoR_gRv2qBNKNFruBY8L3RiIX5lz3M,22458
-sagemaker_core-1.0.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.7.dist-info/METADATA,sha256=v4aXFGELDKUqmf_ZdLBgMzzqulDx2TXKJ_p83nKsxTE,4877
-sagemaker_core-1.0.7.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-sagemaker_core-1.0.7.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.7.dist-info/RECORD,,
+sagemaker_core-1.0.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.8.dist-info/METADATA,sha256=MvlRWwl72_MVfsHjlrirz1RooJ1vJBuyCrZe_HDJr9s,4877
+sagemaker_core-1.0.8.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+sagemaker_core-1.0.8.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.8.dist-info/RECORD,,