`iddlZddlZddlZejejdkZGddeZd dZ d dZ dZ dZ dS) Nz1.10.0c,eZdZdZdZdZdZdZdS)NpzFilec||_dSN)npz_file)selfrs `/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_io/npz.py__init__zNpzFile.__init__ s   c8|j|Sr)r __enter__rs r r zNpzFile.__enter__s !!! r c>|j|||dSr)r__exit__)rtypval tracebacks r rzNpzFile.__exit__s" sC33333r cD|j|}tj|Sr)rcupyarray)rkeyarrs r __getitem__zNpzFile.__getitem__smC z#r c8|jdSr)rclosers r rz NpzFile.closes r N)__name__ __module__ __qualname__r r rrrr r rr s_!!!444r rcftr|dn|}tj|||}n+|tjdtj||}t |tjrtj|St |tj j j rt|S|S)aLoads arrays or pickled objects from ``.npy``, ``.npz`` or pickled file. This function just calls ``numpy.load`` and then sends the arrays to the current device. NPZ file is converted to NpzFile object, which defers the transfer to the time of accessing the items. Args: file (file-like object or string): The file to read. mmap_mode (None, 'r+', 'r', 'w+', 'c'): If not ``None``, memory-map the file to construct an intermediate :class:`numpy.ndarray` object and transfer it to the current device. allow_pickle (bool): Allow loading pickled object arrays stored in npy files. Reasons for disallowing pickles include security, as loading pickled data can execute arbitrary code. If pickles are disallowed, loading object arrays will fail. Please be aware that CuPy does not support arrays with dtype of `object`. The default is False. This option is available only for NumPy 1.10 or later. In NumPy 1.9, this option cannot be specified (loading pickled objects is always allowed). Returns: CuPy array or NpzFile object depending on the type of the file. NpzFile object is a dictionary-like object with the context manager protocol (which enables us to use *with* statement on it). .. seealso:: :func:`numpy.load` NF1allow_pickle option is not supported in NumPy 1.9) _support_allow_picklenumpyloadwarningswarn isinstancendarrayrrlibnpyior)file mmap_mode allow_pickleobjs r r$r$s>* , 4uu, jy,77  # MM N N Njy))#u}%%z# C0 1 1s|| r ctr0|dn|}tj|tj||dS|t jdtj|tj|dS)aSaves an array to a binary file in ``.npy`` format. Args: file (file or str): File or filename to save. arr (array_like): Array to save. It should be able to feed to :func:`cupy.asnumpy`. allow_pickle (bool): Allow saving object arrays using Python pickles. Reasons for disallowing pickles include security (loading pickled data can execute arbitrary code) and portability (pickled objects may not be loadable on different Python installations, for example if the stored objects require libraries that are not available, and not all pickled data is compatible between Python 2 and Python 3). The default is True. This option is available only for NumPy 1.10 or later. In NumPy 1.9, this option cannot be specified (saving objects using pickles is always allowed). .. seealso:: :func:`numpy.save` NTr!)r"r#saverasnumpyr%r&)r+rr-s r r0r0Nsw,,+3tt  4c**L99999  # MM N N N 4c**+++++r cttj|}|D]}tj||||< tj|g|Ri|dS)aTSaves one or more arrays into a file in uncompressed ``.npz`` format. Arguments without keys are treated as arguments with automatic keys named ``arr_0``, ``arr_1``, etc. corresponding to the positions in the argument list. The keys of arguments are used as keys in the ``.npz`` file, which are used for accessing NpzFile object when the file is read by :func:`cupy.load` function. Args: file (file or str): File or filename to save. *args: Arrays with implicit keys. **kwds: Arrays with explicit keys. .. seealso:: :func:`numpy.savez` N)maprr1r#savezr+argskwdsrs r r4r4msc" t|T " "D,,Lc++S  K$t$$$t$$$$$r cttj|}|D]}tj||||< tj|g|Ri|dS)a Saves one or more arrays into a file in compressed ``.npz`` format. It is equivalent to :func:`cupy.savez` function except the output file is compressed. .. seealso:: :func:`cupy.savez` for more detail, :func:`numpy.savez_compressed` N)r3rr1r#savez_compressedr5s r r9r9sd t|T " "D,,Lc++S  4/$///$/////r )NNr) r%r#rr) NumpyVersion __version__r"objectrr$r0r4r9rr r r=s //0ABBhNf(,,,,^,,,,>%%%.00000r