diff --git a/smdebug/rules/action/stop_training_action.py b/smdebug/rules/action/stop_training_action.py index 230096637..c60ac60fb 100644 --- a/smdebug/rules/action/stop_training_action.py +++ b/smdebug/rules/action/stop_training_action.py @@ -79,7 +79,7 @@ def _get_sm_tj_jobs_with_prefix(self): ) break - return found_job_dict.keys() + return list(found_job_dict.keys()) def _stop_training_job(self): if len(self._found_jobs) != 1: