L i1UddlmZddlmZmZddlmZmZddl m Z m Z m Z ddl mZer2ddlmZmZddlmZddlmcmZddlmZmZdd lmZdd lmZdd lm Z n ddlmcmZd#d Z" d$d Z#d%dZ$d&dZ%e jLjNZ'e jPjNZ)ie'jTde'jVde'jXde'jZde'j\de'j^de'j`de'jbde'jdde'jfde'jhde'jjde'jlde'jnde'jpde)jnde)jpdZ9de:d <e'jve'jxe'jze'j|e'j~fZ@d'd!ZAd(d"Zy#e!$rddlZYtwxYw))) annotations) TYPE_CHECKINGAny)make_group_by_kwargsselect_columns_by_name)ImplementationVersionisinstance_or_issubclass) get_pyarrow)MappingSequenceN) DaskLazyFrame Incomplete)DaskExpr)DType) IntoDTypecg}|D]j}||}|j|}t|t|k7rd|dt|d}t||jt ||l|S)NzInternal error: got aliases z, but only got z results)_evaluate_aliaseslenAssertionErrorextendzip)dfexprsnative_resultsexprnative_series_listaliasesmsgs Z/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/narwhals/_dask/utils.pyevaluate_exprsr!s24N@!"X((, w<312 20 M_I`HaaijC % %c'+=>? @ c|Dcgc]=}t|tjr|n|jj |d?c}Scc}w)N)_tmpr$) isinstancedxSeries _native_frameassign)rseriesss r align_series_full_broadcastr,)sO   299 %2+;+;+B+B+B+J6+RR  sAA c |j}|jdi|di}t|jdi|||jddz i|g|tj S)Nblelloch)method)columnsr)rcumsumrDASK)framename original_colsrs r add_row_indexr82smMMM!U\\.T1I.B ! CT2d8??*?=ABC   r"ctjj|j|js d}t |y)NzOObjects are not co-aligned, so this operation is not supported for Dask backend)r&rare_co_aligned_expr RuntimeError)lhsrhsrs r validate_comparandr?<s6 77 ! !#))SYY 7`3 8r"float64float32boolcategoryzdate32[day][pyarrow]int8int16int32int64uint8uint16uint32uint64zdatetime64[us]ztimedelta64[ns]zMapping[type[DType], str]NW_TO_DASK_DTYPESc`|j}|j}tj|x}r|St ||j r0t jjdk\rtrdSdSyt ||jrc|tjur d}t|t||jr!ddl}|j!|j"dSd }t%|t'|t(rd |j*d }t|d |}t-|) N)rrzstring[pyarrow]zstring[python]objectz9Converting to Enum is not supported in narwhals.stable.v1rT)orderedz9Can not cast / initialize Enum without categories presentzConverting to z! dtype is not supported for Dask.zUnknown dtype: )dtypes base_typerLgetr StringrPANDAS_backend_versionr Enumr V1NotImplementedErrorr%pandasCategoricalDtype categories ValueError issubclassUNSUPPORTED_DTYPES__name__r)dtypeversionrQrR dask_typerpds r narwhals_to_native_dtyperejs ^^F!I%)))44y4v}}5  1 1 3y @(3 $ K;K Kv{{3 gjj MC%c* * eV[[ )  &&u'7'7&F FIo)/0y1122ST!#&& E7 #C  r"c@t|}|jd|S)N)drop_null_keysas_index)pd_make_group_by_kwargspop)rgkwargss r rrs $N CF JJz Mr")rrrrreturnzlist[tuple[str, dx.Series]])rrr*zdx.Series | objectrlzSequence[dx.Series])r5 dd.DataFramer6strrlrm)r= dx.Seriesr>rorlNone)rarrbr rlr)rgrBrlzdict[str, bool])B __future__rtypingrrnarwhals._pandas_like.utilsrrirnarwhals._utilsrr r narwhals.dependenciesr collections.abcr r dask.dataframe dataframedddask.dataframe.dask_expr dask_exprr&narwhals._dask.dataframerrnarwhals._dask.exprrnarwhals.dtypesrnarwhals.typingrModuleNotFoundErrorr!r,r8r?MAINrQrX dtypes_v1Float64Float32Boolean CategoricalDateInt8Int16Int32Int64UInt8UInt16UInt32UInt64DatetimeDurationrL__annotations__ListStructArrayTimeBinaryr_rer1r"r rs"%NM-1))B,%)--  2     JJ   0 NNI0 NNI0 NNF0  0  KK' 0  KK 0 LL'0 LL'0 LL'0 LL'0 MM80 MM80 MM80 OO%0 OO&0 (!0")#0,( KK MM LL KK MM :] s G GG