Skip to content

Commit 5334797

Browse files
Add support for SQL compute configuration for StartAudienceGenerationJob API.
1 parent 47685d0 commit 5334797

File tree

7 files changed

+41
-5
lines changed

7 files changed

+41
-5
lines changed

generator/ServiceModels/cleanroomsml/cleanroomsml-2023-09-06.api.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1000,7 +1000,8 @@
10001000
"members":{
10011001
"dataSource":{"shape":"S3ConfigMap"},
10021002
"roleArn":{"shape":"IamRoleArn"},
1003-
"sqlParameters":{"shape":"ProtectedQuerySQLParameters"}
1003+
"sqlParameters":{"shape":"ProtectedQuerySQLParameters"},
1004+
"sqlComputeConfiguration":{"shape":"ComputeConfiguration"}
10041005
}
10051006
},
10061007
"AudienceGenerationJobList":{

generator/ServiceModels/cleanroomsml/cleanroomsml-2023-09-06.docs.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -358,6 +358,7 @@
358358
"ComputeConfiguration": {
359359
"base": "<p>Provides configuration information for the instances that will perform the compute work.</p>",
360360
"refs": {
361+
"AudienceGenerationJobDataSource$sqlComputeConfiguration": null,
361362
"ProtectedQueryInputParameters$computeConfiguration": "<p>Provides configuration information for the workers that will perform the protected query.</p>"
362363
}
363364
},
@@ -1394,7 +1395,7 @@
13941395
"base": "<p>Defines information about the data source used for model inference.</p>",
13951396
"refs": {
13961397
"GetTrainedModelInferenceJobResponse$dataSource": "<p>The data source that was used for the trained model inference job.</p>",
1397-
"StartTrainedModelInferenceJobRequest$dataSource": "<p>Defines he data source that is used for the trained model inference job.</p>"
1398+
"StartTrainedModelInferenceJobRequest$dataSource": "<p>Defines the data source that is used for the trained model inference job.</p>"
13981399
}
13991400
},
14001401
"ModelTrainingDataChannel": {

generator/ServiceModels/cleanroomsml/cleanroomsml-2023-09-06.normal.json

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1094,7 +1094,8 @@
10941094
"sqlParameters":{
10951095
"shape":"ProtectedQuerySQLParameters",
10961096
"documentation":"<p>The protected SQL query parameters.</p>"
1097-
}
1097+
},
1098+
"sqlComputeConfiguration":{"shape":"ComputeConfiguration"}
10981099
},
10991100
"documentation":"<p>Defines the Amazon S3 bucket where the seed audience for the generating audience is stored.</p>"
11001101
},
@@ -5419,7 +5420,7 @@
54195420
},
54205421
"dataSource":{
54215422
"shape":"ModelInferenceDataSource",
5422-
"documentation":"<p>Defines he data source that is used for the trained model inference job.</p>"
5423+
"documentation":"<p>Defines the data source that is used for the trained model inference job.</p>"
54235424
},
54245425
"description":{
54255426
"shape":"ResourceDescription",

sdk/src/Services/CleanRoomsML/Generated/Model/AudienceGenerationJobDataSource.cs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ public partial class AudienceGenerationJobDataSource
3737
{
3838
private S3ConfigMap _dataSource;
3939
private string _roleArn;
40+
private ComputeConfiguration _sqlComputeConfiguration;
4041
private ProtectedQuerySQLParameters _sqlParameters;
4142

4243
/// <summary>
@@ -90,6 +91,21 @@ internal bool IsSetRoleArn()
9091
return this._roleArn != null;
9192
}
9293

94+
/// <summary>
95+
/// Gets and sets the property SqlComputeConfiguration.
96+
/// </summary>
97+
public ComputeConfiguration SqlComputeConfiguration
98+
{
99+
get { return this._sqlComputeConfiguration; }
100+
set { this._sqlComputeConfiguration = value; }
101+
}
102+
103+
// Check to see if SqlComputeConfiguration property is set
104+
internal bool IsSetSqlComputeConfiguration()
105+
{
106+
return this._sqlComputeConfiguration != null;
107+
}
108+
93109
/// <summary>
94110
/// Gets and sets the property SqlParameters.
95111
/// <para>

sdk/src/Services/CleanRoomsML/Generated/Model/Internal/MarshallTransformations/AudienceGenerationJobDataSourceMarshaller.cs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,17 @@ public void Marshall(AudienceGenerationJobDataSource requestObject, JsonMarshall
6565
context.Writer.Write(requestObject.RoleArn);
6666
}
6767

68+
if(requestObject.IsSetSqlComputeConfiguration())
69+
{
70+
context.Writer.WritePropertyName("sqlComputeConfiguration");
71+
context.Writer.WriteObjectStart();
72+
73+
var marshaller = ComputeConfigurationMarshaller.Instance;
74+
marshaller.Marshall(requestObject.SqlComputeConfiguration, context);
75+
76+
context.Writer.WriteObjectEnd();
77+
}
78+
6879
if(requestObject.IsSetSqlParameters())
6980
{
7081
context.Writer.WritePropertyName("sqlParameters");

sdk/src/Services/CleanRoomsML/Generated/Model/Internal/MarshallTransformations/AudienceGenerationJobDataSourceUnmarshaller.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,12 @@ public AudienceGenerationJobDataSource Unmarshall(JsonUnmarshallerContext contex
7878
unmarshalledObject.RoleArn = unmarshaller.Unmarshall(context);
7979
continue;
8080
}
81+
if (context.TestExpression("sqlComputeConfiguration", targetDepth))
82+
{
83+
var unmarshaller = ComputeConfigurationUnmarshaller.Instance;
84+
unmarshalledObject.SqlComputeConfiguration = unmarshaller.Unmarshall(context);
85+
continue;
86+
}
8187
if (context.TestExpression("sqlParameters", targetDepth))
8288
{
8389
var unmarshaller = ProtectedQuerySQLParametersUnmarshaller.Instance;

sdk/src/Services/CleanRoomsML/Generated/Model/StartTrainedModelInferenceJobRequest.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ internal bool IsSetContainerExecutionParameters()
8989
/// <summary>
9090
/// Gets and sets the property DataSource.
9191
/// <para>
92-
/// Defines he data source that is used for the trained model inference job.
92+
/// Defines the data source that is used for the trained model inference job.
9393
/// </para>
9494
/// </summary>
9595
[AWSProperty(Required=true)]

0 commit comments

Comments
 (0)