Fix exception causes in abstract_meta_trainer.py

KezhiAdore · Dec 8, 2022 · 667aa25 · 667aa25
1 parent 49d3036
commit 667aa25
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/open_spiel/python/algorithms/psro_v2/abstract_meta_trainer.py b/open_spiel/python/algorithms/psro_v2/abstract_meta_trainer.py
@@ -47,11 +47,12 @@ def _process_string_or_callable(string_or_callable, dictionary):
 
   try:
     return dictionary[string_or_callable]
-  except KeyError:
+  except KeyError as e:
     raise NotImplementedError("Input type / value not supported. Accepted types"
                               ": string, callable. Acceptable string values : "
                               "{}. Input provided : {}".format(
-                                  list(dictionary.keys()), string_or_callable))
+                                  list(dictionary.keys()),
+                                  string_or_callable)) from e
 
 
 def sample_episode(state, policies):