code error in prior.py?

[orgw]

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