Fixing logic error in training_compiler

Lokiiiiii · Lokiiiiii · commit df4d6d9bc941 · 2022-06-14T23:20:47.000-07:00
diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py
@@ -216,6 +216,7 @@ def __init__(
                 raise ValueError(error_string)
             if compiler_config:
                 compiler_config.validate(self)
+        self.compiler_config = compiler_config
 
     def _validate_args(self, py_version):
         """Placeholder docstring"""
