Function open_array

point_e/evals/npz_stream.py:260–266

(path: str, arr_name: str)

@contextmanager
def open_array(path: str, arr_name: str):
    with open(path, "rb") as f:
        with zipfile.ZipFile(f, "r") as zip_f:
            if f"{arr_name}.npy" not in zip_f.namelist():
                raise ValueError(f"missing {arr_name} in npz file")
            with zip_f.open(f"{arr_name}.npy", "r") as arr_f:
                yield arr_f


def _dict_batch_size(objs: Dict[str, np.ndarray]) -> int: