Bump v0.9.3

kddubey · Aug 6, 2024 · 18b73f0 · 18b73f0
1 parent cf7f6ea
commit 18b73f0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/cappr/__init__.py b/src/cappr/__init__.py
@@ -4,7 +4,7 @@
 https://cappr.readthedocs.io/
 """
 
-__version__ = "0.9.2"
+__version__ = "0.9.3"
 
 from . import utils  # noqa: F401
 from ._example import Example  # noqa: F401

diff --git a/src/cappr/utils/classify.py b/src/cappr/utils/classify.py
@@ -41,7 +41,7 @@ def _setattr(obj, name: str, value):
 def _avg_then_exp(
     log_probs: Sequence[Sequence[float]] | Sequence[Sequence[Sequence[float]]],
     axis: int | None = None,
-) -> npt.NDArray[np.floating]:
+) -> np.floating | npt.NDArray[np.floating]:
     return np.exp(np.mean(log_probs, axis=axis))
 
 
@@ -176,12 +176,12 @@ def agg_log_probs(
     Returns
     -------
     agg: npt.NDArray[np.floating] | list[float] | list[list[float]]
-        If `log_probs` is 2-D, then `agg` is a numpy array or list where::
+        If `log_probs` is 2-D, then `agg` is a numpy array or a list where::
 
             agg[j] = func(log_probs[j])
 
-        If `log_probs` is 3-D, then `agg` is an array or a list of list of probabilities
-        where::
+        If `log_probs` is 3-D, then `agg` is a numpy array or a list of list of
+        probabilities where::
 
             agg[i][j] = func(log_probs[i][j])