def create( cls, action_name=None, source_uri=None, source_type=None, action_type=None, description=None, status=None, properties=None, tags=None, sagemaker_session=None, ): """Create an action and return an ``Action`` object representing it. Args: action_name (str): Name of the action source_uri (str): Source URI of the action source_type (str): Source type of the action action_type (str): The type of the action description (str): Description of the action status (str): Status of the action. properties (dict): key/value properties tags (dict): AWS tags for the action sagemaker_session (sagemaker.session.Session): Session object which manages interactions with Amazon SageMaker APIs and any other AWS services needed. If not specified, one is created using the default AWS configuration chain. Returns: Action: A SageMaker ``Action`` object. """ return super(Action, cls)._construct( cls._boto_create_method, action_name=action_name, source=_api_types.ContextSource(source_uri=source_uri, source_type=source_type), action_type=action_type, description=description, status=status, properties=properties, tags=tags, sagemaker_session=sagemaker_session, )
def create( cls, context_name: str = None, source_uri: str = None, source_type: str = None, context_type: str = None, description: str = None, properties: dict = None, tags: dict = None, sagemaker_session=None, ) -> "Context": """Create a context and return a ``Context`` object representing it. Args: context_name (str): The name of the context. source_uri (str): The source URI of the context. source_type (str): The type of the source. context_type (str): The type of the context. description (str): Description of the context. properties (dict): Metadata associated with the context. tags (dict): Tags to add to the context. sagemaker_session (sagemaker.session.Session): Session object which manages interactions with Amazon SageMaker APIs and any other AWS services needed. If not specified, one is created using the default AWS configuration chain. Returns: Context: A SageMaker ``Context`` object. """ return super(Context, cls)._construct( cls._boto_create_method, context_name=context_name, source=_api_types.ContextSource(source_uri=source_uri, source_type=source_type), context_type=context_type, description=description, properties=properties, tags=tags, sagemaker_session=sagemaker_session, )
def create( cls, artifact_name=None, source_uri=None, source_types=None, artifact_type=None, properties=None, tags=None, sagemaker_session=None, ): """Create an artifact and return an ``Artifact`` object representing it. Args: artifact_name (str, optional): Name of the artifact source_uri (str, optional): Source URI of the artifact source_types (list, optional): Source types artifact_type (str, optional): Type of the artifact properties (dict, optional): key/value properties tags (dict, optional): AWS tags for the artifact sagemaker_session (sagemaker.session.Session): Session object which manages interactions with Amazon SageMaker APIs and any other AWS services needed. If not specified, one is created using the default AWS configuration chain. Returns: Artifact: A SageMaker ``Artifact`` object. """ return super(Artifact, cls)._construct( cls._boto_create_method, artifact_name=artifact_name, source=_api_types.ContextSource(source_uri=source_uri, source_types=source_types), artifact_type=artifact_type, properties=properties, tags=tags, sagemaker_session=sagemaker_session, )
def test_list(sagemaker_session): creation_time = datetime.datetime.now( datetime.timezone.utc) + datetime.timedelta(hours=3) last_modified_time = datetime.datetime.now( datetime.timezone.utc) + datetime.timedelta(hours=4) sagemaker_session.sagemaker_client.list_contexts.side_effect = [ { "ContextSummaries": [{ "ContextName": "A" + str(i), "ContextArn": "B" + str(i), "Source": { "SourceUri": "test-source-uri" + str(i), "SourceType": "test-source-type" + str(i), }, "ContextType": "test-context-type", "CreationTime": creation_time + datetime.timedelta(hours=i), "LastModifiedTime": last_modified_time + datetime.timedelta(hours=i), "LastModifiedBy": {}, } for i in range(10)], "NextToken": "100", }, { "ContextSummaries": [{ "ContextName": "A" + str(i), "ContextArn": "B" + str(i), "Source": { "SourceUri": "test-source-uri" + str(i), "SourceType": "test-source-type" + str(i), }, "ContextType": "test-context-type", "CreationTime": creation_time + datetime.timedelta(hours=i), "LastModifiedTime": last_modified_time + datetime.timedelta(hours=i), "LastModifiedBy": {}, } for i in range(10, 20)] }, ] expected = [ _api_types.ContextSummary( context_name="A" + str(i), context_arn="B" + str(i), source=_api_types.ContextSource( source_uri="test-source-uri" + str(i), source_type="test-source-type" + str(i)), context_type="test-context-type", creation_time=creation_time + datetime.timedelta(hours=i), last_modified_time=last_modified_time + datetime.timedelta(hours=i), last_modified_by={}, ) for i in range(20) ] result = list( context.Context.list( source_uri="foo", sagemaker_session=sagemaker_session, sort_by="CreationTime", sort_order="Ascending", )) assert expected == result expected_calls = [ unittest.mock.call(SortBy="CreationTime", SortOrder="Ascending", SourceUri="foo"), unittest.mock.call(NextToken="100", SortBy="CreationTime", SortOrder="Ascending", SourceUri="foo"), ] assert expected_calls == sagemaker_session.sagemaker_client.list_contexts.mock_calls