Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
477 changes: 467 additions & 10 deletions sample/sagemaker/2017-07-24/service-2.json

Large diffs are not rendered by default.

134 changes: 133 additions & 1 deletion src/sagemaker_core/main/code_injection/shape_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -3082,6 +3082,27 @@
"members": [{"name": "LabelingJobArn", "shape": "LabelingJobArn", "type": "string"}],
"type": "structure",
},
"CreateMlflowAppRequest": {
"members": [
{"name": "Name", "shape": "MlflowAppName", "type": "string"},
{"name": "ArtifactStoreUri", "shape": "S3Uri", "type": "string"},
{"name": "RoleArn", "shape": "RoleArn", "type": "string"},
{"name": "ModelRegistrationMode", "shape": "ModelRegistrationMode", "type": "string"},
{
"name": "WeeklyMaintenanceWindowStart",
"shape": "WeeklyMaintenanceWindowStart",
"type": "string",
},
{"name": "AccountDefaultStatus", "shape": "AccountDefaultStatus", "type": "string"},
{"name": "DefaultDomainIdList", "shape": "DefaultDomainIdList", "type": "list"},
{"name": "Tags", "shape": "TagList", "type": "list"},
],
"type": "structure",
},
"CreateMlflowAppResponse": {
"members": [{"name": "Arn", "shape": "MlflowAppArn", "type": "string"}],
"type": "structure",
},
"CreateMlflowTrackingServerRequest": {
"members": [
{"name": "TrackingServerName", "shape": "TrackingServerName", "type": "string"},
Expand Down Expand Up @@ -3543,6 +3564,22 @@
"members": [{"name": "AuthorizedUrl", "shape": "PresignedDomainUrl", "type": "string"}],
"type": "structure",
},
"CreatePresignedMlflowAppUrlRequest": {
"members": [
{"name": "Arn", "shape": "MlflowAppArn", "type": "string"},
{"name": "ExpiresInSeconds", "shape": "ExpiresInSeconds", "type": "integer"},
{
"name": "SessionExpirationDurationInSeconds",
"shape": "SessionExpirationDurationInSeconds",
"type": "integer",
},
],
"type": "structure",
},
"CreatePresignedMlflowAppUrlResponse": {
"members": [{"name": "AuthorizedUrl", "shape": "MlflowAppUrl", "type": "string"}],
"type": "structure",
},
"CreatePresignedMlflowTrackingServerUrlRequest": {
"members": [
{"name": "TrackingServerName", "shape": "TrackingServerName", "type": "string"},
Expand Down Expand Up @@ -4108,6 +4145,7 @@
"member_type": "structure",
"type": "list",
},
"DefaultDomainIdList": {"member_shape": "DomainId", "member_type": "string", "type": "list"},
"DefaultEbsStorageSettings": {
"members": [
{
Expand Down Expand Up @@ -4380,6 +4418,14 @@
],
"type": "structure",
},
"DeleteMlflowAppRequest": {
"members": [{"name": "Arn", "shape": "MlflowAppArn", "type": "string"}],
"type": "structure",
},
"DeleteMlflowAppResponse": {
"members": [{"name": "Arn", "shape": "MlflowAppArn", "type": "string"}],
"type": "structure",
},
"DeleteMlflowTrackingServerRequest": {
"members": [
{"name": "TrackingServerName", "shape": "TrackingServerName", "type": "string"}
Expand Down Expand Up @@ -5788,6 +5834,34 @@
],
"type": "structure",
},
"DescribeMlflowAppRequest": {
"members": [{"name": "Arn", "shape": "MlflowAppArn", "type": "string"}],
"type": "structure",
},
"DescribeMlflowAppResponse": {
"members": [
{"name": "Arn", "shape": "MlflowAppArn", "type": "string"},
{"name": "Name", "shape": "MlflowAppName", "type": "string"},
{"name": "ArtifactStoreUri", "shape": "S3Uri", "type": "string"},
{"name": "MlflowVersion", "shape": "MlflowVersion", "type": "string"},
{"name": "RoleArn", "shape": "RoleArn", "type": "string"},
{"name": "Status", "shape": "MlflowAppStatus", "type": "string"},
{"name": "ModelRegistrationMode", "shape": "ModelRegistrationMode", "type": "string"},
{"name": "AccountDefaultStatus", "shape": "AccountDefaultStatus", "type": "string"},
{"name": "DefaultDomainIdList", "shape": "DefaultDomainIdList", "type": "list"},
{"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
{"name": "CreatedBy", "shape": "UserContext", "type": "structure"},
{"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
{"name": "LastModifiedBy", "shape": "UserContext", "type": "structure"},
{
"name": "WeeklyMaintenanceWindowStart",
"shape": "WeeklyMaintenanceWindowStart",
"type": "string",
},
{"name": "MaintenanceStatus", "shape": "MaintenanceStatus", "type": "string"},
],
"type": "structure",
},
"DescribeMlflowTrackingServerRequest": {
"members": [
{"name": "TrackingServerName", "shape": "TrackingServerName", "type": "string"}
Expand Down Expand Up @@ -8010,7 +8084,7 @@
"type": "list",
},
"HubContentSearchKeywordList": {
"member_shape": "HubSearchKeyword",
"member_shape": "HubContentSearchKeyword",
"member_type": "string",
"type": "list",
},
Expand Down Expand Up @@ -10315,6 +10389,28 @@
],
"type": "structure",
},
"ListMlflowAppsRequest": {
"members": [
{"name": "CreatedAfter", "shape": "Timestamp", "type": "timestamp"},
{"name": "CreatedBefore", "shape": "Timestamp", "type": "timestamp"},
{"name": "Status", "shape": "MlflowAppStatus", "type": "string"},
{"name": "MlflowVersion", "shape": "MlflowVersion", "type": "string"},
{"name": "DefaultForDomainId", "shape": "String", "type": "string"},
{"name": "AccountDefaultStatus", "shape": "AccountDefaultStatus", "type": "string"},
{"name": "SortBy", "shape": "SortMlflowAppBy", "type": "string"},
{"name": "SortOrder", "shape": "SortOrder", "type": "string"},
{"name": "NextToken", "shape": "NextToken", "type": "string"},
{"name": "MaxResults", "shape": "MaxResults", "type": "integer"},
],
"type": "structure",
},
"ListMlflowAppsResponse": {
"members": [
{"name": "Summaries", "shape": "MlflowAppSummaries", "type": "list"},
{"name": "NextToken", "shape": "NextToken", "type": "string"},
],
"type": "structure",
},
"ListMlflowTrackingServersRequest": {
"members": [
{"name": "CreatedAfter", "shape": "Timestamp", "type": "timestamp"},
Expand Down Expand Up @@ -11359,6 +11455,22 @@
],
"type": "structure",
},
"MlflowAppSummaries": {
"member_shape": "MlflowAppSummary",
"member_type": "structure",
"type": "list",
},
"MlflowAppSummary": {
"members": [
{"name": "Arn", "shape": "MlflowAppArn", "type": "string"},
{"name": "Name", "shape": "MlflowAppName", "type": "string"},
{"name": "Status", "shape": "MlflowAppStatus", "type": "string"},
{"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
{"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
{"name": "MlflowVersion", "shape": "MlflowVersion", "type": "string"},
],
"type": "structure",
},
"Model": {
"members": [
{"name": "ModelName", "shape": "ModelName", "type": "string"},
Expand Down Expand Up @@ -16625,6 +16737,26 @@
],
"type": "structure",
},
"UpdateMlflowAppRequest": {
"members": [
{"name": "Arn", "shape": "MlflowAppArn", "type": "string"},
{"name": "Name", "shape": "MlflowAppName", "type": "string"},
{"name": "ArtifactStoreUri", "shape": "S3Uri", "type": "string"},
{"name": "ModelRegistrationMode", "shape": "ModelRegistrationMode", "type": "string"},
{
"name": "WeeklyMaintenanceWindowStart",
"shape": "WeeklyMaintenanceWindowStart",
"type": "string",
},
{"name": "DefaultDomainIdList", "shape": "DefaultDomainIdList", "type": "list"},
{"name": "AccountDefaultStatus", "shape": "AccountDefaultStatus", "type": "string"},
],
"type": "structure",
},
"UpdateMlflowAppResponse": {
"members": [{"name": "Arn", "shape": "MlflowAppArn", "type": "string"}],
"type": "structure",
},
"UpdateMlflowTrackingServerRequest": {
"members": [
{"name": "TrackingServerName", "shape": "TrackingServerName", "type": "string"},
Expand Down
4 changes: 4 additions & 0 deletions src/sagemaker_core/main/config_schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -461,6 +461,10 @@
},
},
},
"MlflowApp": {
"type": "object",
"properties": {"role_arn": {"type": "string"}},
},
"MlflowTrackingServer": {
"type": "object",
"properties": {"role_arn": {"type": "string"}},
Expand Down
Loading
Loading