o
    `Û·is  ã                   @   sÊ   d dl Z d dlmZ ee jƒedƒkr ee jƒedƒk r edƒ‚ddlmZmZmZm	Z	 ddl
mZ dd	lmZmZmZmZ e jZd
d„ Zeje_e jjjZdd„ Zeje_dd„ Zeeeƒ g d¢ZdS )é    N)ÚVersionÚ0z	2024.11.0z\Dask on Ray requires Dask version 2024.11.0 or later. Please upgrade your Dask installation.é   )ÚProgressBarCallbackÚRayDaskCallbackÚlocal_ray_callbacksÚunpack_ray_callbacks)Údataframe_optimize)Údisable_dask_on_rayÚenable_dask_on_rayÚray_dask_getÚray_dask_get_syncc                  O   s   d|d< t | i |¤ŽS ©NTÚray_persist)Údask_persist)ÚargsÚkwargs© r   úL/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/util/dask/__init__.pyÚray_dask_persist   s   r   c                 K   s   d|d< t | fi |¤ŽS r   )Údask_persist_mixin)Úselfr   r   r   r   Úray_dask_persist_mixin)   s   r   c                 C   s   | t _|t jj_d S )N)ÚdaskÚpersistÚbaseÚDaskMethodsMixin)r   r   r   r   r   Ú
patch_dask4   s   r   )
r   r
   r   r   r   r   r   r   r	   r   )r   Úpackaging.versionr   Ú__version__ÚImportErrorÚ	callbacksr   r   r   r   Úoptimizationsr	   Ú	schedulerr
   r   r   r   r   r   r   Ú__doc__r   r   r   r   r   Ú__all__r   r   r   r   Ú<module>   s(    ÿÿ

