Try updating syntax so coverage isn't broken; if it does not work try…

… passing in --ignore-errors=True
infer-actively · Jun 12, 2024 · a0b2ccd · a0b2ccd
1 parent bb0c6cd
commit a0b2ccd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymdp/jax/task.py b/pymdp/jax/task.py
@@ -9,9 +9,9 @@
 
 def select_probs(positions, matrix, dependency_list, actions=None):
     args = tuple(p for i, p in enumerate(positions) if i in dependency_list)
-    args += () if actions is None else (actions,)
+    args = args + (actions,) if actions is not None else args
 
-    return matrix[..., *args]
+    return matrix[(...,) + args]
 
 def cat_sample(key, p):
     a = jnp.arange(p.shape[-1])