o
    c²“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mZmZmZ dd	l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.é   )Úray_dask_getÚray_dask_get_syncÚenable_dask_on_rayÚdisable_dask_on_ray)ÚRayDaskCallbackÚlocal_ray_callbacksÚunpack_ray_callbacksÚProgressBarCallback)Údataframe_optimizec                  O   s   d|d< t | i |¤ŽS ©NTÚray_persist)Údask_persist)ÚargsÚkwargs© r   úJ/home/ubuntu/.local/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Ú	schedulerr   r   r   r   Ú	callbacksr	   r
   r   r   Úoptimizationsr   r   r   r   Ú__doc__r   r   r   r   r   Ú__all__r   r   r   r   Ú<module>   s(    ÿÿ

