def to_pickle(obj, path, compression="infer", protocol=4): if isinstance(obj, DataFrame): obj = obj._query_compiler return BaseFactory.to_pickle(obj, path, compression=compression, protocol=protocol)
def to_pickle( obj: Any, filepath_or_buffer: Union[str, pathlib.Path], compression: Optional[str] = "infer", protocol: int = 4, ): if isinstance(obj, DataFrame): obj = obj._query_compiler return BaseFactory.to_pickle( obj, filepath_or_buffer, compression=compression, protocol=protocol )