blub

dirmeier · Feb 22, 2024 · f8d40cc · f8d40cc
1 parent f8b5f7f
commit f8d40cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sbijax/_src/_sbi_base.py b/sbijax/_src/_sbi_base.py
@@ -19,7 +19,7 @@ def __init__(self, model_fns):
         self._len_theta = len(self.prior_sampler_fn(seed=jr.PRNGKey(123)))
 
     @abc.abstractmethod
-    def sample_posterior(self, rng_key: jr.KeyArray, **kwargs):
+    def sample_posterior(self, rng_key, **kwargs):
         """Sample from the posterior distribution.
 
         Args: