try not creating array from pkl_data["xs"]