Python EngineDispatcher.to_pickle 예제들

예제 #1

파일: io.py 프로젝트: rolveb/modin

def to_pickle(
    obj: Any,
    filepath_or_buffer: Union[str, pathlib.Path],
    compression: Optional[str] = "infer",
    protocol: int = 4,
):
    from modin.data_management.factories.dispatcher import EngineDispatcher

    if isinstance(obj, DataFrame):
        obj = obj._query_compiler
    return EngineDispatcher.to_pickle(obj,
                                      filepath_or_buffer,
                                      compression=compression,
                                      protocol=protocol)

예제 #2

파일: io.py 프로젝트: yangl235/modin

def to_pickle(
    obj: Any,
    filepath_or_buffer: Union[str, pathlib.Path],
    compression: Optional[str] = "infer",
    protocol: int = pickle.HIGHEST_PROTOCOL,
    storage_options: StorageOptions = None,
):
    from modin.data_management.factories.dispatcher import EngineDispatcher

    Engine.subscribe(_update_engine)
    if isinstance(obj, DataFrame):
        obj = obj._query_compiler
    return EngineDispatcher.to_pickle(obj,
                                      filepath_or_buffer,
                                      compression=compression,
                                      protocol=protocol)