Raise KeyError instead of AssertionError

Junyoungpark · Nov 24, 2021 · 40cb34b · 40cb34b
1 parent 51212d9
commit 40cb34b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optuna/integration/allennlp/_variables.py b/optuna/integration/allennlp/_variables.py
@@ -47,7 +47,8 @@ def prefix(self) -> str:
 
     def _get_key(self, name: str) -> str:
         key = self.NAME_OF_KEY.get(name)
-        assert key is not None, f"{name} is not found in `{self.NAME_OF_PATH}`."
+        if key is None:
+            raise KeyError(f"{name} is not found in `{self.NAME_OF_PATH}`.")
         return key
 
     def set_value(self, name: str, value: str) -> None: