fix pylint format errors

Author: yijiezh

src/sagemaker/rl/estimator.py

@@ -42,8 +42,8 @@
         "0.5": {"tensorflow": "1.11"},
         "0.6.5": {"tensorflow": "1.12"},
         "0.6": {"tensorflow": "1.12"},
-        "0.8.2":{"tensorflow": "2.1"},
-        "0.8.5":{"tensorflow": "2.1", "pytorch": "1.5"}
+        "0.8.2": {"tensorflow": "2.1"},
+        "0.8.5": {"tensorflow": "2.1", "pytorch": "1.5"},
     },
 }
 
@@ -146,7 +146,9 @@ def __init__(
         self._validate_images_args(toolkit, toolkit_version, framework, image_name)
 
         if not image_name:
-            self._validate_toolkit_support(toolkit.value, toolkit_version, framework.value)
+            self._validate_toolkit_support(
+                toolkit.value, toolkit_version, framework.value
+            )
             self.toolkit = toolkit.value
             self.toolkit_version = toolkit_version
             self.framework = framework.value
@@ -263,10 +265,14 @@ def create_model(
             return tfsModel(framework_version=self.framework_version, **base_args)
         if self.framework == RLFramework.MXNET.value:
             return MXNetModel(
-                framework_version=self.framework_version, py_version=PYTHON_VERSION, **extended_args
+                framework_version=self.framework_version,
+                py_version=PYTHON_VERSION,
+                **extended_args
             )
         raise ValueError(
-            "An unknown RLFramework enum was passed in. framework: {}".format(self.framework)
+            "An unknown RLFramework enum was passed in. framework: {}".format(
+                self.framework
+            )
         )
 
     def train_image(self):
@@ -290,8 +296,8 @@ def train_image(self):
                 self.train_instance_type,
                 self._image_version(),
                 py_version="py36",
-                account=DEFAULT_RL_ACCOUNT
-             )
+                account=DEFAULT_RL_ACCOUNT,
+            )
 
         return fw_utils.create_image_uri(
             self.sagemaker_session.boto_region_name,
@@ -302,7 +308,9 @@ def train_image(self):
         )
 
     @classmethod
-    def _prepare_init_params_from_job_description(cls, job_details, model_channel_name=None):
+    def _prepare_init_params_from_job_description(
+        cls, job_details, model_channel_name=None
+    ):
         """Convert the job description to init params that can be handled by the
         class constructor
 
@@ -356,7 +364,9 @@ def hyperparameters(self):
             SAGEMAKER_ESTIMATOR: SAGEMAKER_ESTIMATOR_VALUE,
         }
 
-        hyperparameters.update(Framework._json_encode_hyperparameters(additional_hyperparameters))
+        hyperparameters.update(
+            Framework._json_encode_hyperparameters(additional_hyperparameters)
+        )
         return hyperparameters
 
     @classmethod
@@ -394,7 +404,9 @@ def _validate_toolkit_format(cls, toolkit):
         """
         if toolkit and toolkit not in list(RLToolkit):
             raise ValueError(
-                "Invalid type: {}, valid RL toolkits types are: {}".format(toolkit, list(RLToolkit))
+                "Invalid type: {}, valid RL toolkits types are: {}".format(
+                    toolkit, list(RLToolkit)
+                )
             )
 
     @classmethod
@@ -506,7 +518,15 @@ def default_metric_definitions(cls, toolkit):
             float_regex = "[-+]?[0-9]*[.]?[0-9]+([eE][-+]?[0-9]+)?"  # noqa: W605, E501
 
             return [
-                {"Name": "episode_reward_mean", "Regex": "episode_reward_mean: (%s)" % float_regex},
-                {"Name": "episode_reward_max", "Regex": "episode_reward_max: (%s)" % float_regex},
+                {
+                    "Name": "episode_reward_mean",
+                    "Regex": "episode_reward_mean: (%s)" % float_regex,
+                },
+                {
+                    "Name": "episode_reward_max",
+                    "Regex": "episode_reward_max: (%s)" % float_regex,
+                },
             ]
-        raise ValueError("An unknown RLToolkit enum was passed in. toolkit: {}".format(toolkit))
+        raise ValueError(
+            "An unknown RLToolkit enum was passed in. toolkit: {}".format(toolkit)
+        )