Trip initilialisation of SBR of unbias=True, since it's incompatible

dynamicslab · Jan 23, 2024 · 28dd0f3 · 28dd0f3
1 parent 6e9b003
commit 28dd0f3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pysindy/optimizers/sbr.py b/pysindy/optimizers/sbr.py
@@ -81,9 +81,15 @@ def __init__(
         num_warmup: int = 1000,
         num_samples: int = 5000,
         mcmc_kwargs: Optional[dict] = None,
+        unbias: bool = False,
         **kwargs,
     ):
-        super().__init__(**kwargs)
+
+        if unbias:
+            raise ValueError("SBR is incompatible with unbiasing. Set unbias=False")
+
+        super().__init__(unbias=unbias, **kwargs)
+
         # set the hyperparameters
         self.sparsity_coef_tau0 = sparsity_coef_tau0
         self.slab_shape_nu = slab_shape_nu