We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce6f566 commit ae3abd4Copy full SHA for ae3abd4
src/sagemaker/workflow/airflow.py
@@ -649,9 +649,7 @@ def model_config_from_estimator(
649
model = estimator.create_model(vpc_config_override=vpc_config_override)
650
elif isinstance(estimator, TensorFlow):
651
model = estimator.create_model(
652
- role=role,
653
- vpc_config_override=vpc_config_override,
654
- entry_point=estimator.entry_point,
+ role=role, vpc_config_override=vpc_config_override, entry_point=estimator.entry_point
655
)
656
elif isinstance(estimator, sagemaker.estimator.Framework):
657
0 commit comments