Skip to content

Commit 0615e41

Browse files
authored
Update estimator.py
1 parent f0f94c0 commit 0615e41

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

src/sagemaker/rl/estimator.py

+2-8
Original file line numberDiff line numberDiff line change
@@ -506,13 +506,7 @@ def default_metric_definitions(cls, toolkit):
506506
float_regex = "[-+]?[0-9]*[.]?[0-9]+([eE][-+]?[0-9]+)?" # noqa: W605, E501
507507

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

0 commit comments

Comments
 (0)