Skip to content

Commit 690bd0d

Browse files
add: recovery mode for SageMaker Studio apps
1 parent 64edba2 commit 690bd0d

File tree

7 files changed

+63
-1
lines changed

7 files changed

+63
-1
lines changed

generator/ServiceModels/sagemaker/sagemaker-2017-07-24.api.json

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7259,7 +7259,8 @@
72597259
"AppType":{"shape":"AppType"},
72607260
"AppName":{"shape":"AppName"},
72617261
"Tags":{"shape":"TagList"},
7262-
"ResourceSpec":{"shape":"ResourceSpec"}
7262+
"ResourceSpec":{"shape":"ResourceSpec"},
7263+
"RecoveryMode":{"shape":"Boolean"}
72637264
}
72647265
},
72657266
"CreateAppResponse":{
@@ -9837,6 +9838,7 @@
98379838
"UserProfileName":{"shape":"UserProfileName"},
98389839
"SpaceName":{"shape":"SpaceName"},
98399840
"Status":{"shape":"AppStatus"},
9841+
"RecoveryMode":{"shape":"Boolean"},
98409842
"LastHealthCheckTimestamp":{"shape":"Timestamp"},
98419843
"LastUserActivityTimestamp":{"shape":"Timestamp"},
98429844
"CreationTime":{"shape":"Timestamp"},

generator/ServiceModels/sagemaker/sagemaker-2017-07-24.docs.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1634,13 +1634,15 @@
16341634
"ChannelSpecification$IsRequired": "<p>Indicates whether the channel is required by the algorithm.</p>",
16351635
"ClarifyCheckStepMetadata$SkipCheck": "<p>This flag indicates if the drift check against the previous baseline will be skipped or not. If it is set to <code>False</code>, the previous baseline of the configured check type must be available.</p>",
16361636
"ClarifyCheckStepMetadata$RegisterNewBaseline": "<p>This flag indicates if a newly calculated baseline can be accessed through step properties <code>BaselineUsedForDriftCheckConstraints</code> and <code>BaselineUsedForDriftCheckStatistics</code>. If it is set to <code>False</code>, the previous baseline of the configured check type must also be available. These can be accessed through the <code>BaselineUsedForDriftCheckConstraints</code> property. </p>",
1637+
"CreateAppRequest$RecoveryMode": "<p> Indicates whether the application is launched in recovery mode. </p>",
16371638
"CreateEndpointConfigInput$EnableNetworkIsolation": "<p>Sets whether all model containers deployed to the endpoint are isolated. If they are, no inbound or outbound network calls can be made to or from the model containers.</p>",
16381639
"CreateMlflowTrackingServerRequest$AutomaticModelRegistration": "<p>Whether to enable or disable automatic registration of new MLflow models to the SageMaker Model Registry. To enable automatic model registration, set this value to <code>True</code>. To disable automatic model registration, set this value to <code>False</code>. If not specified, <code>AutomaticModelRegistration</code> defaults to <code>False</code>.</p>",
16391640
"CreateModelInput$EnableNetworkIsolation": "<p>Isolates the model container. No inbound or outbound network calls can be made to or from the model container.</p>",
16401641
"CreatePartnerAppRequest$EnableIamSessionBasedIdentity": "<p>When set to <code>TRUE</code>, the SageMaker Partner AI App sets the Amazon Web Services IAM session name or the authenticated IAM user as the identity of the SageMaker Partner AI App user.</p>",
16411642
"CreateTrainingJobRequest$EnableNetworkIsolation": "<p>Isolates the training container. No inbound or outbound network calls can be made, except for calls between peers within a training cluster for distributed training. If you enable network isolation for training jobs that are configured to use a VPC, SageMaker downloads and uploads customer data and model artifacts through the specified VPC, but the training container does not have network access.</p>",
16421643
"CreateTrainingJobRequest$EnableInterContainerTrafficEncryption": "<p>To encrypt all communications between ML compute instances in distributed training, choose <code>True</code>. Encryption provides greater security for distributed training, but training might take longer. How long it takes depends on the amount of communication between compute instances, especially if you use a deep learning algorithm in distributed training. For more information, see <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/train-encrypt.html\">Protect Communications Between ML Compute Instances in a Distributed Training Job</a>.</p>",
16431644
"CreateTrainingJobRequest$EnableManagedSpotTraining": "<p>To train models using managed spot training, choose <code>True</code>. Managed spot training provides a fully managed and scalable infrastructure for training machine learning models. this option is useful when training jobs can be interrupted and when there is flexibility when the training job is run. </p> <p>The complete and intermediate results of jobs are stored in an Amazon S3 bucket, and can be used as a starting point to train models incrementally. Amazon SageMaker provides metrics and logs in CloudWatch. They can be used to see when managed spot training jobs are running, interrupted, resumed, or completed. </p>",
1645+
"DescribeAppResponse$RecoveryMode": "<p> Indicates whether the application is launched in recovery mode. </p>",
16441646
"DescribeEndpointConfigOutput$EnableNetworkIsolation": "<p>Indicates whether all model containers deployed to the endpoint are isolated. If they are, no inbound or outbound network calls can be made to or from the model containers.</p>",
16451647
"DescribeMlflowTrackingServerResponse$AutomaticModelRegistration": "<p>Whether automatic registration of new MLflow models to the SageMaker Model Registry is enabled.</p>",
16461648
"DescribeModelOutput$EnableNetworkIsolation": "<p>If <code>True</code>, no inbound or outbound network calls can be made to or from the model container.</p>",

generator/ServiceModels/sagemaker/sagemaker-2017-07-24.normal.json

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9161,6 +9161,10 @@
91619161
"ResourceSpec":{
91629162
"shape":"ResourceSpec",
91639163
"documentation":"<p>The instance type and the Amazon Resource Name (ARN) of the SageMaker AI image created on the instance.</p> <note> <p>The value of <code>InstanceType</code> passed as part of the <code>ResourceSpec</code> in the <code>CreateApp</code> call overrides the value passed as part of the <code>ResourceSpec</code> configured for the user profile or the domain. If <code>InstanceType</code> is not specified in any of those three <code>ResourceSpec</code> values for a <code>KernelGateway</code> app, the <code>CreateApp</code> call fails with a request validation error.</p> </note>"
9164+
},
9165+
"RecoveryMode":{
9166+
"shape":"Boolean",
9167+
"documentation":"<p> Indicates whether the application is launched in recovery mode. </p>"
91649168
}
91659169
}
91669170
},
@@ -13888,6 +13892,10 @@
1388813892
"shape":"AppStatus",
1388913893
"documentation":"<p>The status.</p>"
1389013894
},
13895+
"RecoveryMode":{
13896+
"shape":"Boolean",
13897+
"documentation":"<p> Indicates whether the application is launched in recovery mode. </p>"
13898+
},
1389113899
"LastHealthCheckTimestamp":{
1389213900
"shape":"Timestamp",
1389313901
"documentation":"<p>The timestamp of the last health check.</p>"

sdk/src/Services/SageMaker/Generated/Model/CreateAppRequest.cs

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ public partial class CreateAppRequest : AmazonSageMakerRequest
4141
private string _appName;
4242
private AppType _appType;
4343
private string _domainId;
44+
private bool? _recoveryMode;
4445
private ResourceSpec _resourceSpec;
4546
private string _spaceName;
4647
private List<Tag> _tags = AWSConfigs.InitializeCollections ? new List<Tag>() : null;
@@ -103,6 +104,24 @@ internal bool IsSetDomainId()
103104
return this._domainId != null;
104105
}
105106

107+
/// <summary>
108+
/// Gets and sets the property RecoveryMode.
109+
/// <para>
110+
/// Indicates whether the application is launched in recovery mode.
111+
/// </para>
112+
/// </summary>
113+
public bool RecoveryMode
114+
{
115+
get { return this._recoveryMode.GetValueOrDefault(); }
116+
set { this._recoveryMode = value; }
117+
}
118+
119+
// Check to see if RecoveryMode property is set
120+
internal bool IsSetRecoveryMode()
121+
{
122+
return this._recoveryMode.HasValue;
123+
}
124+
106125
/// <summary>
107126
/// Gets and sets the property ResourceSpec.
108127
/// <para>

sdk/src/Services/SageMaker/Generated/Model/DescribeAppResponse.cs

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ public partial class DescribeAppResponse : AmazonWebServiceResponse
4343
private string _failureReason;
4444
private DateTime? _lastHealthCheckTimestamp;
4545
private DateTime? _lastUserActivityTimestamp;
46+
private bool? _recoveryMode;
4647
private ResourceSpec _resourceSpec;
4748
private string _spaceName;
4849
private AppStatus _status;
@@ -226,6 +227,24 @@ internal bool IsSetLastUserActivityTimestamp()
226227
return this._lastUserActivityTimestamp.HasValue;
227228
}
228229

230+
/// <summary>
231+
/// Gets and sets the property RecoveryMode.
232+
/// <para>
233+
/// Indicates whether the application is launched in recovery mode.
234+
/// </para>
235+
/// </summary>
236+
public bool RecoveryMode
237+
{
238+
get { return this._recoveryMode.GetValueOrDefault(); }
239+
set { this._recoveryMode = value; }
240+
}
241+
242+
// Check to see if RecoveryMode property is set
243+
internal bool IsSetRecoveryMode()
244+
{
245+
return this._recoveryMode.HasValue;
246+
}
247+
229248
/// <summary>
230249
/// Gets and sets the property ResourceSpec.
231250
/// <para>

sdk/src/Services/SageMaker/Generated/Model/Internal/MarshallTransformations/CreateAppRequestMarshaller.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,12 @@ public IRequest Marshall(CreateAppRequest publicRequest)
8787
context.Writer.Write(publicRequest.DomainId);
8888
}
8989

90+
if(publicRequest.IsSetRecoveryMode())
91+
{
92+
context.Writer.WritePropertyName("RecoveryMode");
93+
context.Writer.Write(publicRequest.RecoveryMode);
94+
}
95+
9096
if(publicRequest.IsSetResourceSpec())
9197
{
9298
context.Writer.WritePropertyName("ResourceSpec");

sdk/src/Services/SageMaker/Generated/Model/Internal/MarshallTransformations/DescribeAppResponseUnmarshaller.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,12 @@ public override AmazonWebServiceResponse Unmarshall(JsonUnmarshallerContext cont
106106
response.LastUserActivityTimestamp = unmarshaller.Unmarshall(context);
107107
continue;
108108
}
109+
if (context.TestExpression("RecoveryMode", targetDepth))
110+
{
111+
var unmarshaller = BoolUnmarshaller.Instance;
112+
response.RecoveryMode = unmarshaller.Unmarshall(context);
113+
continue;
114+
}
109115
if (context.TestExpression("ResourceSpec", targetDepth))
110116
{
111117
var unmarshaller = ResourceSpecUnmarshaller.Instance;

0 commit comments

Comments
 (0)