solved #178 (#179)

ugr-sail · Mar 15, 2022 · 07ed0dd · 07ed0dd
1 parent 0c02db3
commit 07ed0dd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/sinergym/utils/callbacks.py b/sinergym/utils/callbacks.py
@@ -90,7 +90,11 @@ def _on_step(self) -> bool:
             try:
                 action = self.locals['action'][-1]
             except KeyError:
-                print('Algorithm action key in locals dict unknown')
+                try:
+                    action = self.locals['actions'][-1]
+                except KeyError:
+                    raise KeyError(
+                        'Algorithm action key in locals dict unknown.')
 
         if self.training_env.get_attr('flag_discrete')[0]:
             action = self.training_env.get_attr('action_mapping')[0][action]

diff --git a/sinergym/version.txt b/sinergym/version.txt
@@ -1 +1 @@
-1.7.6
+1.7.7