workaround for np.random.choice returning (1,N) array