1// Code generated by smithy-go-codegen DO NOT EDIT.
2
3package glue
4
5import (
6	"context"
7	awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
8	"github.com/aws/aws-sdk-go-v2/aws/signer/v4"
9	"github.com/aws/aws-sdk-go-v2/service/glue/types"
10	"github.com/aws/smithy-go/middleware"
11	smithyhttp "github.com/aws/smithy-go/transport/http"
12)
13
14// Updates an existing machine learning transform. Call this operation to tune the
15// algorithm parameters to achieve better results. After calling this operation,
16// you can call the StartMLEvaluationTaskRun operation to assess how well your new
17// parameters achieved your goals (such as improving the quality of your machine
18// learning transform, or making it more cost-effective).
19func (c *Client) UpdateMLTransform(ctx context.Context, params *UpdateMLTransformInput, optFns ...func(*Options)) (*UpdateMLTransformOutput, error) {
20	if params == nil {
21		params = &UpdateMLTransformInput{}
22	}
23
24	result, metadata, err := c.invokeOperation(ctx, "UpdateMLTransform", params, optFns, addOperationUpdateMLTransformMiddlewares)
25	if err != nil {
26		return nil, err
27	}
28
29	out := result.(*UpdateMLTransformOutput)
30	out.ResultMetadata = metadata
31	return out, nil
32}
33
34type UpdateMLTransformInput struct {
35
36	// A unique identifier that was generated when the transform was created.
37	//
38	// This member is required.
39	TransformId *string
40
41	// A description of the transform. The default is an empty string.
42	Description *string
43
44	// This value determines which version of AWS Glue this machine learning transform
45	// is compatible with. Glue 1.0 is recommended for most customers. If the value is
46	// not set, the Glue compatibility defaults to Glue 0.9. For more information, see
47	// AWS Glue Versions
48	// (https://docs.aws.amazon.com/glue/latest/dg/release-notes.html#release-notes-versions)
49	// in the developer guide.
50	GlueVersion *string
51
52	// The number of AWS Glue data processing units (DPUs) that are allocated to task
53	// runs for this transform. You can allocate from 2 to 100 DPUs; the default is 10.
54	// A DPU is a relative measure of processing power that consists of 4 vCPUs of
55	// compute capacity and 16 GB of memory. For more information, see the AWS Glue
56	// pricing page (https://aws.amazon.com/glue/pricing/). When the WorkerType field
57	// is set to a value other than Standard, the MaxCapacity field is set
58	// automatically and becomes read-only.
59	MaxCapacity *float64
60
61	// The maximum number of times to retry a task for this transform after a task run
62	// fails.
63	MaxRetries *int32
64
65	// The unique name that you gave the transform when you created it.
66	Name *string
67
68	// The number of workers of a defined workerType that are allocated when this task
69	// runs.
70	NumberOfWorkers *int32
71
72	// The configuration parameters that are specific to the transform type (algorithm)
73	// used. Conditionally dependent on the transform type.
74	Parameters *types.TransformParameters
75
76	// The name or Amazon Resource Name (ARN) of the IAM role with the required
77	// permissions.
78	Role *string
79
80	// The timeout for a task run for this transform in minutes. This is the maximum
81	// time that a task run for this transform can consume resources before it is
82	// terminated and enters TIMEOUT status. The default is 2,880 minutes (48 hours).
83	Timeout *int32
84
85	// The type of predefined worker that is allocated when this task runs. Accepts a
86	// value of Standard, G.1X, or G.2X.
87	//
88	// * For the Standard worker type, each worker
89	// provides 4 vCPU, 16 GB of memory and a 50GB disk, and 2 executors per worker.
90	//
91	// *
92	// For the G.1X worker type, each worker provides 4 vCPU, 16 GB of memory and a
93	// 64GB disk, and 1 executor per worker.
94	//
95	// * For the G.2X worker type, each worker
96	// provides 8 vCPU, 32 GB of memory and a 128GB disk, and 1 executor per worker.
97	WorkerType types.WorkerType
98}
99
100type UpdateMLTransformOutput struct {
101
102	// The unique identifier for the transform that was updated.
103	TransformId *string
104
105	// Metadata pertaining to the operation's result.
106	ResultMetadata middleware.Metadata
107}
108
109func addOperationUpdateMLTransformMiddlewares(stack *middleware.Stack, options Options) (err error) {
110	err = stack.Serialize.Add(&awsAwsjson11_serializeOpUpdateMLTransform{}, middleware.After)
111	if err != nil {
112		return err
113	}
114	err = stack.Deserialize.Add(&awsAwsjson11_deserializeOpUpdateMLTransform{}, middleware.After)
115	if err != nil {
116		return err
117	}
118	if err = addSetLoggerMiddleware(stack, options); err != nil {
119		return err
120	}
121	if err = awsmiddleware.AddClientRequestIDMiddleware(stack); err != nil {
122		return err
123	}
124	if err = smithyhttp.AddComputeContentLengthMiddleware(stack); err != nil {
125		return err
126	}
127	if err = addResolveEndpointMiddleware(stack, options); err != nil {
128		return err
129	}
130	if err = v4.AddComputePayloadSHA256Middleware(stack); err != nil {
131		return err
132	}
133	if err = addRetryMiddlewares(stack, options); err != nil {
134		return err
135	}
136	if err = addHTTPSignerV4Middleware(stack, options); err != nil {
137		return err
138	}
139	if err = awsmiddleware.AddRawResponseToMetadata(stack); err != nil {
140		return err
141	}
142	if err = awsmiddleware.AddRecordResponseTiming(stack); err != nil {
143		return err
144	}
145	if err = addClientUserAgent(stack); err != nil {
146		return err
147	}
148	if err = smithyhttp.AddErrorCloseResponseBodyMiddleware(stack); err != nil {
149		return err
150	}
151	if err = smithyhttp.AddCloseResponseBodyMiddleware(stack); err != nil {
152		return err
153	}
154	if err = addOpUpdateMLTransformValidationMiddleware(stack); err != nil {
155		return err
156	}
157	if err = stack.Initialize.Add(newServiceMetadataMiddleware_opUpdateMLTransform(options.Region), middleware.Before); err != nil {
158		return err
159	}
160	if err = addRequestIDRetrieverMiddleware(stack); err != nil {
161		return err
162	}
163	if err = addResponseErrorMiddleware(stack); err != nil {
164		return err
165	}
166	if err = addRequestResponseLogging(stack, options); err != nil {
167		return err
168	}
169	return nil
170}
171
172func newServiceMetadataMiddleware_opUpdateMLTransform(region string) *awsmiddleware.RegisterServiceMetadata {
173	return &awsmiddleware.RegisterServiceMetadata{
174		Region:        region,
175		ServiceID:     ServiceID,
176		SigningName:   "glue",
177		OperationName: "UpdateMLTransform",
178	}
179}
180