Backwards compatability

tingiskhan · Nov 1, 2024 · 9193a71 · 9193a71
1 parent ec7f9bb
commit 9193a71
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/numpyro_sts/base.py b/numpyro_sts/base.py
@@ -1,3 +1,4 @@
+import warnings
 from functools import cached_property, reduce
 from typing import Tuple
 
@@ -90,7 +91,12 @@ def __init__(
         std_is_matrix: bool = False,
         column_mask: np.ndarray = None,
         validate_args=None,
+        **kwargs,
     ):
+        if "mask" in kwargs:
+            warnings.warn("'mask' is deprecated in favor of 'column_mask'", DeprecationWarning)
+            column_mask = kwargs.pop("mask")
+
         _verify_parameters(offset, matrix, std, initial_value, std_is_matrix)
         times = jnp.arange(n)