Merge remote-tracking branch 'origin/fast_sbr_test' into simultaneous

dynamicslab · Oct 16, 2024 · e4b9b95 · e4b9b95
2 parents de77d21 + 4b15699
commit e4b9b95
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pysindy/_typing.py b/pysindy/_typing.py
@@ -0,0 +1,7 @@
+import numpy as np
+import numpy.typing as npt
+
+# In python 3.12, use type statement
+# https://docs.python.org/3/reference/simple_stmts.html#the-type-statement
+NpFlt = np.floating[npt.NBitBase]
+Float2D = np.ndarray[tuple[int, int], np.dtype[NpFlt]]
diff --git a/test/test_optimizers/test_optimizers.py b/test/test_optimizers/test_optimizers.py
@@ -126,7 +126,7 @@ def data(request):
         ElasticNet(fit_intercept=False),
         DummyLinearModel(),
         MIOSR(),
-        SBR(),
+        SBR(num_warmup=10, num_samples=10),
     ],
     ids=lambda param: type(param),
 )