Fixed bug in type hints for "_sample_reg_horseshoe"

dynamicslab · Feb 12, 2024 · b2579bb · b2579bb
1 parent ed358cb
commit b2579bb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pysindy/optimizers/sbr.py b/pysindy/optimizers/sbr.py
@@ -1,4 +1,5 @@
 from typing import Optional
+from typing import Tuple
 
 import jax.numpy as jnp
 import numpy as np
@@ -181,7 +182,7 @@ def _run_mcmc(self, x, y, **kwargs):
         return mcmc
 
 
-def _sample_reg_horseshoe(tau: float, c_sq: float, shape: tuple[int, ...]):
+def _sample_reg_horseshoe(tau: float, c_sq: float, shape: Tuple[int, ...]):
     """Create a regularized horseshoe distribution
 
     The regularized horseshoe distribution behaves like a horseshoe prior when