Skip to content

Commit b038ecb

Browse files
authored
Expose TuningStep
The TuningStep Class is not exposed, making it not possible to import like other steps. e.g. `from stepfunctions.steps import TuningStep`
1 parent 6404067 commit b038ecb

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
19+
from stepfunctions.steps.sagemaker import TrainingStep, TransformStep, ModelStep, EndpointConfigStep, EndpointStep, TuningStep
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)