reformatting

Author: icywang86rui

src/sagemaker/tensorflow/estimator.py

@@ -190,7 +190,7 @@ class TensorFlow(Framework):
 
     __framework_name__ = "tensorflow"
 
-    LATEST_VERSION = '1.13'
+    LATEST_VERSION = "1.13"
     """The latest version of TensorFlow included in the SageMaker pre-built Docker images."""
 
     _LOWEST_SCRIPT_MODE_ONLY_VERSION = [1, 13]
@@ -324,11 +324,15 @@ def _validate_args(
                 )
 
         if (not self._script_mode_enabled()) and self._only_script_mode_supported():
-            logger.warning('Legacy mode is deprecated in versions 1.13 and higher. Using script mode instead.')
+            logger.warning(
+                "Legacy mode is deprecated in versions 1.13 and higher. Using script mode instead."
+            )
             self.script_mode = True
 
     def _only_script_mode_supported(self):
-        return [int(s) for s in self.framework_version.split('.')] >= self._LOWEST_SCRIPT_MODE_ONLY_VERSION
+        return [
+            int(s) for s in self.framework_version.split(".")
+        ] >= self._LOWEST_SCRIPT_MODE_ONLY_VERSION
 
     def _validate_requirements_file(self, requirements_file):
         if not requirements_file:

src/sagemaker/utils.py

@@ -132,7 +132,7 @@ def get_short_version(framework_version):
     Returns:
         str: The short version string
     """
-    return '.'.join(framework_version.split('.')[:2])
+    return ".".join(framework_version.split(".")[:2])
 
 
 def to_str(value):

tests/unit/test_tf_estimator.py

@@ -871,12 +871,20 @@ def test_script_mode_deprecated_args(sagemaker_session):
 
 
 def test_legacy_mode_deprecated(sagemaker_session):
-    tf = _build_tf(sagemaker_session=sagemaker_session, framework_version='1.13.1',
-                   py_version='py2', script_mode=False)
+    tf = _build_tf(
+        sagemaker_session=sagemaker_session,
+        framework_version="1.13.1",
+        py_version="py2",
+        script_mode=False,
+    )
     assert tf._script_mode_enabled() is True
 
-    tf = _build_tf(sagemaker_session=sagemaker_session, framework_version='1.12',
-                   py_version='py2', script_mode=False)
+    tf = _build_tf(
+        sagemaker_session=sagemaker_session,
+        framework_version="1.12",
+        py_version="py2",
+        script_mode=False,
+    )
     assert tf._script_mode_enabled() is False
 
 

tests/unit/test_utils.py

@@ -48,8 +48,8 @@ def test_get_config_value():
 
 
 def test_get_short_version():
-    assert sagemaker.utils.get_short_version('1.13.1') == '1.13'
-    assert sagemaker.utils.get_short_version('1.13') == '1.13'
+    assert sagemaker.utils.get_short_version("1.13.1") == "1.13"
+    assert sagemaker.utils.get_short_version("1.13") == "1.13"
 
 
 def test_deferred_error():