Skip to content

Commit 7b34574

Browse files
committed
feature: Adding Training Compiler support for TensorFlow estimator
1 parent e64718c commit 7b34574

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/sagemaker/tensorflow/estimator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
2828
from sagemaker.tensorflow.training_compiler.config import TrainingCompilerConfig
2929
from sagemaker.workflow import is_pipeline_variable
30+
from sagemaker.tensorflow.training_compiler.config import TrainingCompilerConfig
3031

3132
logger = logging.getLogger("sagemaker")
3233

0 commit comments

Comments
 (0)