i am trying to sample with num_atoms= 'stat'. i suppose this is only possible in beta_prior? this part raises an error(commented). Is this correct? # armsca_res = [d - r for d, r in zip(armsca_distances, arm_stds.numpy())] i fixed it to arm_stds = np.array([arm_stds[0, 0].item(), arm_stds[1, 0].item()]).reshape(1, 2) armsca_res = [d - r for d, r in zip(armsca_distances, arm_stds)] and it works
i am trying to sample with num_atoms= 'stat'. i suppose this is only possible in beta_prior?
this part raises an error(commented). Is this correct?
i fixed it to
and it works