Extend Error handling in run_inference_algorithm

Minor change.
blackjax-devs · Dec 14, 2023 · 7499bfd · 7499bfd
1 parent f12fc38
commit 7499bfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/blackjax/util.py b/blackjax/util.py
@@ -182,7 +182,7 @@ def run_inference_algorithm(
     init_key, sample_key = split(rng_key, 2)
     try:
         initial_state = inference_algorithm.init(initial_state_or_position, init_key)
-    except TypeError:
+    except (TypeError, ValueError, AttributeError):
         # We assume initial_state is already in the right format.
         initial_state = initial_state_or_position