turn action into integer when indexing into slices of B array in `upd…

…ate_state_likelihood_dirichlet()`, to avoid improper index error
infer-actively · Apr 1, 2022 · fcf1152 · fcf1152
1 parent ea83bca
commit fcf1152
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymdp/learning.py b/pymdp/learning.py
@@ -100,7 +100,7 @@ def update_state_likelihood_dirichlet(
 
     for factor in factors:
         dfdb = maths.spm_cross(qs[factor], qs_prev[factor])
-        dfdb *= (B[factor][:, :, actions[factor]] > 0).astype("float")
+        dfdb *= (B[factor][:, :, int(actions[factor])] > 0).astype("float")
         qB[factor][:,:,int(actions[factor])] += (lr*dfdb)
 
     return qB