o $is@sddlZddlmZeejedkr eejedkr edddlmZmZmZm Z ddl m Z dd l m Z mZmZmZejZd d Zeje_ejjjZd d Zeje_ddZeeegdZdS)N)Version0z 2024.11.0z\Dask on Ray requires Dask version 2024.11.0 or later. Please upgrade your Dask installation.)ProgressBarCallbackRayDaskCallbacklocal_ray_callbacksunpack_ray_callbacks)dataframe_optimize)disable_dask_on_rayenable_dask_on_ray ray_dask_getray_dask_get_synccOsd|d<t|i|SNT ray_persist) dask_persist)argskwargsrS/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/util/dask/__init__.pyray_dask_persistsrcKsd|d<t|fi|Sr)dask_persist_mixin)selfrrrrray_dask_persist_mixin)srcCs|t_|tjj_dS)N)daskpersistbaseDaskMethodsMixin)rrrrr patch_dask4sr) r r r r rrrrr r)rpackaging.versionr __version__ ImportError callbacksrrrr optimizationsr schedulerr r r r rrr__doc__rrrrr__all__rrrrs(