Skip to content

Commit ae3abd4

Browse files
committed
black format
1 parent ce6f566 commit ae3abd4

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/sagemaker/workflow/airflow.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -649,9 +649,7 @@ def model_config_from_estimator(
649649
model = estimator.create_model(vpc_config_override=vpc_config_override)
650650
elif isinstance(estimator, TensorFlow):
651651
model = estimator.create_model(
652-
role=role,
653-
vpc_config_override=vpc_config_override,
654-
entry_point=estimator.entry_point,
652+
role=role, vpc_config_override=vpc_config_override, entry_point=estimator.entry_point
655653
)
656654
elif isinstance(estimator, sagemaker.estimator.Framework):
657655
model = estimator.create_model(

0 commit comments

Comments
 (0)