bidZddlmZddlZddlmZddlmZedZ edZ edZ ed Z ed Z ed Zed Zed ZedZedZedZedZedZedZedZedZedZedZ ddZdZddZdZd dZd!dZ y)"z4 Dispatch in dask.dataframe. Also see extension.py ) annotationsN)is_dask_collection)Dispatchmake_meta_dispatch make_meta_obj meta_nonemptymeta_lib_from_arrayhash_object_dispatchgroup_split_dispatchget_parallel_typeCategoricalDtypeconcattolistis_categorical_dtypeunion_categoricalsgrouperpartd_encode_dispatchpyarrow_schema_dispatchfrom_pyarrow_table_dispatchto_pyarrow_table_dispatchto_pandas_dispatchc t|dk(r|dStjt|d}||f|||||d|S)aConcatenate, handling some edge cases: - Unions categoricals between partitions - Ignores empty partitions Parameters ---------- dfs : list of DataFrame, Series, or Index axis : int or str, optional join : str, optional uniform : bool, optional Whether to treat ``dfs[0]`` as representative of ``dfs[1:]``. Set to True if all arguments have the same columns and dtypes (but not necessarily categories). Default is False. ignore_index : bool, optional Whether to allow index values to be ignored/dropped during concatenation. Default is False. ignore_order : bool, optional Whether to ignore the order when doing the union of categoricals. Default is False. r)axisjoinuniformfilter_warning ignore_index)lenconcat_dispatchdispatchtype)dfsrrrrrkwargsfuncs R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/dataframe/dispatch.pyrr"s^< 3x1}1v ''SV 5  )%    cht|d|}tjt|}||S)Ndtype)getattris_categorical_dtype_dispatchr!r"objr%s r&rrOs. #w $C ( 1 1$s) rPs ##23) ) 45 67 6701%&898$8$ ()? @&';<I& !89"#<=&'DE$%@A23   * Z 8  ((VVr'