fix cache=True if r is larger than what's cached

jorenham · Jul 3, 2023 · f150b4a · f150b4a
1 parent 0113cf7
commit f150b4a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/lmo/_lm.py b/lmo/_lm.py
@@ -193,8 +193,10 @@ def l_weights(
         if P_r.shape[0] < r:
             P_r = P_r[:r]
 
-        assert P_r.shape == (r, n)
-        return cast(npt.NDArray[T], P_r)
+        # ignore if r is larger that what's cached
+        if P_r.shape[0] == r:
+            assert P_r.shape == (r, n)
+            return cast(npt.NDArray[T], P_r)
 
 
     if sum(trim) == 0:

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "lmo"
-version = "0.6.0"
+version = "0.6.1"
 description = "L-Moments for robust statistics."
 authors = ["Joren Hammudoglu <[email protected]>"]
 readme = "README.md"