From fcf11528936e8c08ad0519b724a51deb61e9713c Mon Sep 17 00:00:00 2001 From: conorheins Date: Fri, 1 Apr 2022 17:57:40 +0200 Subject: [PATCH] turn action into integer when indexing into slices of B array in `update_state_likelihood_dirichlet()`, to avoid improper index error --- pymdp/learning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymdp/learning.py b/pymdp/learning.py index bd694a6d..ec334f68 100644 --- 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