thanks! does the positional arguments means that in pm.CustomDist, I should pass the parameter the same as in the dist function function, so there I should pass mu instead of mu[cond_idx,subj_idx,time_idx]?
if so, but there the mu would change according to the dims, so maybe I should construct CustomDist in some alternative ways? do you have any advice?