i ddlmZddlZddlZddlZddlmZmZddlZddl m Z ddl Z erddl mZddlmZddl mZmZd dZd d Zd dd Zy)) annotationsN) TYPE_CHECKINGAny)import_optional_dependency)Callable)Path) DataFrameSeriescXtj||tj|S)a Pickle an object and then read it again. Parameters ---------- obj : any object The object to pickle and then re-read. path : str, path object or file-like object, default None The path where the pickled object is written and then read. Returns ------- pandas object The original object that was pickled and then re-read. )pd to_pickle read_pickle)objtmp_paths h/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/_testing/_io.pyround_trip_picklers! LLh >>( ##c&||||}|S)a Write an object to file specified by a pathlib.Path and read it back Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. )writerreaderrrs rround_trip_pathlibr1s$ 8  C Jrc2|f}d}d}|dk(rtj}d}||f}d}n|dk(rTtj}d}tj|}t j |} t||_|| f}d}nk|d k(rtj}nU|d k(rd d l } | j}n?|d k(rtdj}n$|dk(rd d l} | j }nt#d||||5} t%| ||d d d y #1swYy xYw)a Write data to a compressed file. Parameters ---------- compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd'} The compression type to use. path : str The file path to write the data. data : str The data to write. dest : str, default "test" The destination file (for ZIP only) Raises ------ ValueError : An invalid compression value was passed in. wbwritezipwwritestrtar)nameaddfilegzipbz2rNzstd zstandardxzzUnrecognized compression type: )mode)zipfileZipFiletarfileTarFileTarInfoioBytesIOlensizer"GzipFiler#BZ2FileropenlzmaLZMAFile ValueErrorgetattr) compressionpathdatadestargsr'methodcompress_methodfilebytesr#r4fs rwrite_to_compressedrBHs&"GD D Fe!//d|  !//D) 4 I e}  --  ++  4[AFF  --:;-HII D )Q6D! * ) )s 4D  D)rrrrreturnzDataFrame | Series)rr)test)r9strr;rErCNone) __future__rr"r-r*typingrrr(pandas.compat._optionalrpandasr collections.abcrpathlibrr r rrrBrrrrMs>" >($(.9"r