change format data for .npz instead of .pkl