Skip to content

Commit af57234

Browse files
committed
fix format errors
1 parent 1bf8c66 commit af57234

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

src/sagemaker/rl/estimator.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@
4242
"0.5": {"tensorflow": "1.11"},
4343
"0.6.5": {"tensorflow": "1.12"},
4444
"0.6": {"tensorflow": "1.12"},
45-
"0.8.2":{"tensorflow": "2.1"},
46-
"0.8.5":{"tensorflow": "2.1", "pytorch": "1.5"}
45+
"0.8.2": {"tensorflow": "2.1"},
46+
"0.8.5": {"tensorflow": "2.1", "pytorch": "1.5"},
4747
},
4848
}
4949

@@ -290,8 +290,8 @@ def train_image(self):
290290
self.train_instance_type,
291291
self._image_version(),
292292
py_version="py36",
293-
account=DEFAULT_RL_ACCOUNT
294-
)
293+
account=DEFAULT_RL_ACCOUNT,
294+
)
295295

296296
return fw_utils.create_image_uri(
297297
self.sagemaker_session.boto_region_name,
@@ -506,7 +506,10 @@ def default_metric_definitions(cls, toolkit):
506506
float_regex = "[-+]?[0-9]*[.]?[0-9]+([eE][-+]?[0-9]+)?" # noqa: W605, E501
507507

508508
return [
509-
{"Name": "episode_reward_mean", "Regex": "episode_reward_mean: (%s)" % float_regex},
510-
{"Name": "episode_reward_max", "Regex": "episode_reward_max: (%s)" % float_regex},
509+
{
510+
"Name": "episode_reward_mean",
511+
"Regex": "episode_reward_mean: (%s)" % float_regex,
512+
},
513+
{"Name": "episode_reward_max", "Regex": "episode_reward_max: (%s)" % float_regex, },
511514
]
512515
raise ValueError("An unknown RLToolkit enum was passed in. toolkit: {}".format(toolkit))

0 commit comments

Comments
 (0)