Skip to content

Commit bc72e3f

Browse files
authored
Merge pull request #77 from mriccia/patch-1
Expose TuningStep from SageMaker steps
2 parents 2ba55b4 + c1f4485 commit bc72e3f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/stepfunctions/steps/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
from stepfunctions.steps.states import Pass, Succeed, Fail, Wait, Choice, Parallel, Map, Task, Chain, Retry, Catch
1818
from stepfunctions.steps.states import Graph, FrozenGraph
19-
from stepfunctions.steps.sagemaker import TrainingStep, TransformStep, ModelStep, EndpointConfigStep, EndpointStep, ProcessingStep
19+
from stepfunctions.steps.sagemaker import TrainingStep, TransformStep, ModelStep, EndpointConfigStep, EndpointStep, TuningStep, ProcessingStep
2020
from stepfunctions.steps.compute import LambdaStep, BatchSubmitJobStep, GlueStartJobRunStep, EcsRunTaskStep
2121
from stepfunctions.steps.service import DynamoDBGetItemStep, DynamoDBPutItemStep, DynamoDBUpdateItemStep, DynamoDBDeleteItemStep
2222
from stepfunctions.steps.service import SnsPublishStep, SqsSendMessageStep

0 commit comments

Comments
 (0)