-PhUddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddlmZddlmZerGddlZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!eeefZ"d e#d<edefZ$d e#d<n& ddl%m cm&Z'n#e($rddl&Z'YnwxYwe'j)jZdZ*d e#d< ddZ+ddZ,ddZ-Gddedde*fZ.dS) ) annotations)partial) TYPE_CHECKINGAnyCallableClassVarMappingSequenceN)DepthTrackingGroupBy)!evaluate_output_names_and_aliases)GroupBy) SeriesGroupBy) TypeAlias)NarwhalsAggregation DaskLazyFrame)DaskExprrPandasSeriesGroupBy._AggFnz str | _AggFn Aggregationreturndd.Aggregationc@d d}d d}tjd|| S) Nsrrpd.Series[Any]c.|dS)NF)dropna)nunique)rs W/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/narwhals/_dask/group_by.pychunkzn_unique..chunk$syyy&&&s0c*|SN)sum)r"s raggzn_unique..agg'svvxxr!r)namer r&)rrrr)r"rrr)ddr)r r&s rn_uniquer)#sC'''' >y3 ? ? ??r!ddofintc8ttj|SNr*)r _DaskGroupByvarr.s rr0r0- <#$ / / //r!c8ttj|Sr-)rr/stdr.s rr3r31r1r!c DeZdZUdddddeededd Zd ed <ddZddZ dS)DaskLazyGroupByr%meanmedianmaxminsizecount) r%r6r7r8r9r3r0lenr)r;z3ClassVar[Mapping[NarwhalsAggregation, Aggregation]] _REMAP_AGGSdfrkeys"Sequence[DaskExpr] | Sequence[str]drop_null_keysboolrNonec|||\|_|_|_|jj|j|d|_dS)N)r?T)robserved) _parse_keys_compliant_frame_keys_output_key_names compliantnativegroupby_grouped)selfr>r?rAs r__init__zDaskLazyGroupBy.__init__CshEIDTDT TEUE E Atz4+A-55 J~6   r!exprsrc ddlm}|sf|jj|j|jdtt|j|j S| |i}g|j|j R}|D]}t||j|\}}|j dkrX|jd}| |j |t|| f~| || t% r  di|jn | fdt||D||jjdi||jj|jjtt|j|j S)Nrrany)keepc3*K|] \}}||ffVdSr$).0alias output_nameagg_fns r z&DaskLazyGroupBy.agg..rsE''&E;f-.''''''r!)backend_versionversionrU)narwhals._dask.dataframerrJ simple_selectrHuniquerenamedictziprI_ensure_all_simpler _depth_remap_expr_name_function_nameupdatefromkeys _leaf_namecallable_scalar_kwargsrMr& reset_index_backend_version_version) rNrPrsimple_aggregationsexcludeexpr output_namesaliasescolumnrYs @rr&zDaskLazyGroupBy.aggRs8:::::: -,dj9 //ST-CDDEEFF  &&&CE8DJ8!788  D$Edng%% !L'{aA..t/BCC#**4==66BR+S+STTT**4??4+@+@AAF6>v6F6FRVV22d1222FF  & &''''*-g|*D*D'''    } DM  4 4 3 4 4 @ @ B B N;N+    &c$*d&<==>> ? ?  @r!N)r>rr?r@rArBrrC)rPrrr) __name__ __module__ __qualname__r3r0r)r=__annotations__rOr&rUr!rr5r55s H HK         (@(@(@(@(@(@r!r5rr)rr)r*r+rr)/ __future__r functoolsrtypingrrrrr r dask.dataframe dataframer(narwhals._compliantr narwhals._expression_parsingr pandaspddask.dataframe.apir r/pandas.core.groupbyr_PandasSeriesGroupBytyping_extensionsrnarwhals._compliant.group_byrr]rnarwhals._dask.exprrrrxrdask.dataframe.dask_expr dask_exprdxModuleNotFoundError_groupbyrr)r0r3r5rUr!rrsM"""""""LLLLLLLLLLLLLLLL444444JJJJJJ'::::::IIIIII++++++@@@@@@666666,,,,,,%9#s(%CCCCC c*F*****---------- ;&L' ''''B@@@@00000000E@E@E@E@E@*?J +STE@E@E@E@E@s: B BB