def artifact_objs(sagemaker_session): artifact_objs = [] for artifact_name in names(): artifact_objs.append( artifact.Artifact.create( artifact_name=artifact_name, artifact_type="SDKIntegrationTest", source_uri=name(), properties={"k1": "v1"}, sagemaker_session=sagemaker_session, )) time.sleep(SLEEP_TIME_SECONDS) artifact_objs.append( artifact.Artifact.create( artifact_name=name(), artifact_type="SDKIntegrationTestType2", source_uri=name(), properties={"k1": "v1"}, sagemaker_session=sagemaker_session, )) yield artifact_objs for artifact_obj in artifact_objs: artifact_obj.delete()
def context_objs(sagemaker_session): context_objs = [] for context_name in names(): context_objs.append( context.Context.create( context_name=context_name, context_type="SDKIntegrationTest", source_uri="foo", properties={"k1": "v1"}, sagemaker_session=sagemaker_session, )) time.sleep(SLEEP_TIME_SECONDS) yield context_objs for context_obj in context_objs: context_obj.delete()
def action_objs(sagemaker_session): action_objs = [] for action_name in names(): action_objs.append( action.Action.create( action_name=action_name, action_type="SDKIntegrationTest", source_uri="foo", status="InProgress", properties={"k1": "v1"}, sagemaker_session=sagemaker_session, )) time.sleep(SLEEP_TIME_SECONDS) yield action_objs for action_obj in action_objs: action_obj.delete()