CR Fix

pybandits/base.py

@@ -87,7 +87,9 @@ def __init__(self, **data: Any) -> None:
     @validate_arguments
     def _make_epsilon_greedy(epsilon: Float01, default_action: Optional[ActionId], select_action: Callable):
         """
-        Wraps a select_action function with epsilon-greedy strategy.
+        Wraps a select_action function with epsilon-greedy strategy, such that with probability epsilon a default_action
+        is selected, and with probability 1-epsilon the select_action function is triggered to choose action.
+        If no default_action is provided, a random action is selected.
 
         Parameters
         ----------
@@ -102,6 +104,11 @@ def _make_epsilon_greedy(epsilon: Float01, default_action: Optional[ActionId], s
         -------
         select_epsilon_greedy_action: Callable
             The wrapped function.
+
+        Raises
+        ------
+        KeyError
+            If default_action is not present as a key in the probabilities dictionary.
         """
 
         def select_epsilon_greedy_action(