fix missing default

pyro-ppl · Nov 12, 2024 · 3ac0e61 · 3ac0e61
1 parent b24e301
commit 3ac0e61
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/numpyro/contrib/control_flow/scan.py b/numpyro/contrib/control_flow/scan.py
@@ -287,6 +287,9 @@ def scan_wrapper(
     if length is None:
         length = jnp.shape(jax.tree.flatten(xs)[0][0])[0]
 
+    if substitute_stack is None:
+        substitute_stack = []
+
     if enum and history > 0:
         return scan_enum(  # TODO: replay for enum
             f,