AHj`PUddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZmZddlZdd lmZ dd l!m"Z"m#Z#m$Z$dd l!m%Z&dd l'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4er]ddl m5Z5m6Z6m7Z7m8Z8ddlm9Z9m:Z:m;Z;ddlmm?Z?m@Z@ejdk\rddlmBZBnddlCmBZBe9dZDedZEGdde:ZFejdk\rCijijijgZJdeKd<eLdeJDZMeNZOdeKd< dQ dRdZPdSd ZQdTd!ZRd"d"d# dUd$ZSd"d% dVd&ZTd"d% dWd'ZUd"d% dXd(ZVdYd)ZWd"d"d# dZd*ZXd[d+ZYd\d,ZZ dQ d]d-Z[d^d.Z\d_d/Z]d_d0Z^d`d1Z_d2d3dad4Z`dbd5Zadcd6Zbddd7Zceded8Zdedfd9Zddgd:Zd dhd;Zeed<ZfGd=d>egZhejd?ZjGd@dAeZkekjZmeekjZn dQddddB didCZo djdDZpd_dEZqd"dF dkdGZrdldHZsd2dd"dIdJ dmdKZtd"dLdndMZudodNZvGdOdPZwy)p) annotationsN)Counter) Collection GeneratorIterable MappingViewSequenceSized)Enum)BytesIO)Path) TYPE_CHECKINGAnyLiteralNoReturnTypeVaroverload) functions)_check_for_numpyimport_optional subprocess)numpy)find_stacklevel)BooleanDateDatetimeDecimalDurationInt64StringTime) FLOAT_DTYPESINTEGER_DTYPES)CallableIteratorMutableMapping Reversible) ParamSpecProtocol TypeGuard) DataFrameExpr)PolarsDataTypeSizeUnit) )TypeIsPTceZdZddZy)IdentityFunctioncyN)selffns M/root/tools/cai/cai_env/lib/python3.12/site-packages/polars/_utils/various.py__call__zIdentityFunction.__call__DsN)r:Callable[P, T]returnr>)__name__ __module__ __qualname__r<r8r=r;r5r5BsHr=r5r/ zlist[Reversible[Any]]_viewsc#DK|]}tt|ywr7)typereversed).0views r; rKJs"KD4#7"Ks r_OmittedcXt|trt|jS|Sr7) isinstancedictlistitems) null_valuess r;_process_null_valuesrSPs(+t$K%%'((r=ct|ttfxrt|t xs7t|txs%t j dk\xrt|tS)NrC)rNrrr rsys version_info_reverse_mapping_views)vals r; _is_generatorrYYsY C)X. / N 38N4N c; '     ' 83 67 r=c,tfd|DS)z9Check whether the given iterable is of the given type(s).c36K|]}t|ywr7rN)rIxeltypes r;rKz"_is_iterable_of..fs2z!V$2s)all)rXr^s `r;_is_iterable_ofr`ds 2c2 22r=F) allow_strinclude_seriesc|durt|tryt|rHt|tjr.tj |j tjS|r7t|tjr|j tjk(St|t xr(t|txrt|ttfS)z Check that `val` is a sequence of strings or paths. Note that a single string is a sequence of strings by definition, use `allow_str=False` to return False on a single string. F)rNstrrnpndarray issubdtypedtypestr_plSeriesr bytesr r`r rXrarbs r;is_path_or_str_sequencernisEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% sE "" . sH % . C$ -r=)rbc<t|r7t|tjr|jtj k(S|r7t|t jr|jt jk(St|txrt|tS)z;Check whether the given sequence is a sequence of booleans.) rrNrerfrhbool_rjrkrr r`boolrXrbs r;is_bool_sequencersshC!<yyBHH$$ JsBII6yyBJJ&& c8 $ Cd)CCr=cXt|rHt|tjr.tj|j tj S|r4t|tjr|j jSt|txrt|tS)z;Check whether the given sequence is a sequence of integers.) rrNrerfrgrhintegerrjrk is_integerr r`intrrs r;is_int_sequencerxsnC!<}}SYY 33 JsBII6yy##%% c8 $ Bc)BBr=ct|xrt|tjxs;t||rtj t fnt xrt|t S)zBCheck whether the given input is a numpy array or python sequence.)rrNrerfrjrkr rdrrs r; is_sequencerzsO S ! Ajbjj&A 3H-XN %3$ $r=crttt|tx}tuxr t ||Sr7)rqnextiter NO_DEFAULTrN)objtpfirsts r;is_non_empty_sequence_ofrs6 tCy*- -j@ " ub ! r=c|durt|tryt|rHt|tjr.tj |j tjS|r7t|tjr|j tjk(St|txrt|tS)z Check that `val` is a sequence of strings. Note that a single string is a sequence of strings by definition, use `allow_str=False` to return False on a single string. F) rNrdrrerfrgrhrirjrkr r r`rms r;is_str_sequencersEjc2 # :c2::#>}}SYY00 JsBII6yyBII%% c8 $ Bc)BBr=c^ddlm}t||xr|jj S)z9Indicate if the given object is a basic/unaliased column.r)r,) polars.exprr,rNmeta is_column)rr,s r;rrs$ c4 9SXX%7%7%99r=cR|%tjdttyy)z6Warn for possibly unintentional comparisons with None.Nz]Comparisons with None always result in null. Consider using `.is_null()` or `.is_not_null()`.) stacklevel)warningswarn UserWarningrrs r;warn_null_comparisonrs$ { k &( r=cV|xst}|jr9tj|j|j |j |d}nFtj|j|j |j dj|}|j|S)z/Fast conversion of the given range to a Series.T)startendsteprheager)rrrr) rrvF int_rangerstoprcastalias)namerngrhranges r;range_to_seriesrs NUE  ))T  )) $u+  ;;t r=cXt|j|j|jS)z.Return the given range as an equivalent slice.)slicerrr)rs r;range_to_slicers CHHchh //r=ct ddlm}|x}rd|jvryy#t$rYyt$rYywxYw)Nr) get_ipython IPKernelAppFT)IPythonrconfig ImportErrorAttributeError)ripys r; _in_notebookrsL '= C 3::-   s! 777cJ ddl}|jS#t$rYywxYw)NrF)marimorunning_in_notebookr)mos r;_in_marimo_notebookrs+%%'' s  ""cr t|tttfrdSt |S#t $rYywxYw)zLReturn length of (non-string/dict) sequence; returns None for non-sequences.N)rNrdrlrOlen TypeErrorrs r;arrlenrs8!#UD'9:tHCH s* * 66T)check_not_directoryctjj|}|rOtjj|r0tjj |rd|d}t ||S)z>Create a string path, expanding the home directory if present.zexpected a file path; z is a directory)ospath expanduserexistsisdirIsADirectoryError)rrmsgs r;normalize_filepathrsY 77  d #D GGNN4 GGMM$ &tho>$$ Kr=cht|tr|jd}td|DS)zASimple version parser; split into a tuple of ints for comparison..c 3nK|]-}ttjddt|/yw)z\DN)rwresubrd)rIvs r;rKz parse_version..s&AARVVE2s1v./As35)rNrdsplittuple)versions r; parse_versionr s+'3--$ AA AAr=ctt}|j}|Dcgc]}||vr||r|c}Scc}w)zMReturn unique list of sequence values, maintaining their order of appearance.)setadd)valuesseenadd_rs r;ordered_uniquers2UD 88D brl>kb kilobytesi>mb megabytesi>gb gigabytesi@>tb terabyteslz9`unit` must be one of {'b', 'kb', 'mb', 'gb', 'tb'}, got ) ValueError)rrrs r;rr-sq ~ $ $Dy $ $G| $ $G| $ $G|KD8Tor=c  |js8|jjD]}|tk7s d|}t |hd}d}dddd||dz|dzdz|dzd z|dzd zd zd d)fd }i}|j D]\}}| |j tk(r t|j} tj|jjdd} tj| jjdk(j| dzj!| dzjj#ddjj%| d||<t'|dd ||j(j+|j,||<,|t.k(r4tj|jj%|d||<i|t0k(rtjtj|jjdk(jtj|dzj!tj|dzjj#ddjj%|d||<8|t2k(rVtj|j5|t6j9t3dj9|||<|t:k(r,tj|j=ddd||<|t>vrtj|jjAd d} tCj| jjdkDj| j9|||<V|tDvs|j tFk(r+tj|jjd!d"} tj|jjd!d#} tCjtj|jjId$tj|jjKjM|zjtCjtj|jjdkDjtj|jtj|jjId%jtj|jjd&d'j!tCjN| jjAd d| d'(j9tj9|||<||j|k7stj|j9|||<|r|jPd*i|S|S)+a Utility function to cast table repr/string values into frame-native types. Parameters ---------- df Dataframe containing string-repr column data. schema DataFrame schema containing the desired end-state types. Notes ----- Table repr strings are less strict (or different) than equivalent CSV data, so need special handling; as this function is only used for reprs, parsing is flexible. z6DataFrame should contain only String repr data; found >+inf-infinfnaniʚ;rii@B<i) nsusuµsmssmhdwc\|dStfdtjd|DS)Nc3bK|]&\}}t||jz(ywr7)rwstrip)rIvaluerduration_scalings r;rKzH_cast_repr_strings_with_schema..str_duration_..ls2E4E -djjl;;s,/z([+-]?\d+)(\D+))sumrfindall)tdrs r; str_duration_z5_cast_repr_strings_with_schema..str_duration_hs<z  #%::.@"#E r=Nz[A-Z ]+$rz .000000000 000000000rz%Y-%m-%d %H:%M:%S.%9f time_zonez%Y-%m-%dz %H:%M:%S.%9f) return_dtyperTF)truefalsez[^\d+-]z ^(.*)\D(\d*)$z$1z$2z ^[+-]?\d*$z[eE]z [^eE\d+-]r) separator)r str | Noner? int | Noner8))is_emptyschemarr rrQ base_typer time_unitrcolrdreplacewhen len_bytesthen otherwiserstrptimegetattrdtreplace_time_zonerrr!r map_elementsrrrreplace_strictr# replace_allrjr"rcontains to_lowercaseis_in concat_str with_columns)dfr rrspecial_floatsns_secr cast_colsctp_baser int_string integer_partfractional_partrs @r;_cast_repr_strings_with_schemar+>s& ;;=))""$ %BV|NrfUn$ % 4NF  b[ b[2  e^b e^b 1 $  I@12 >||~)"2<<0EE!HLL((b9FF155??,23T!l*+Yq;/Sq"S'+BC ! 2{D1=#,Q?DDQUU1XNOT!%%(,,//78T!%%(,,..|SABY (,,88RH+&)T&\T"X'! *ryy|# uuQx}}R0 ! A@1D,5?2?? 'Y '<"  , K s%%99N)r1rr2ztype[NS]r?r,)r@rArBr3r8r=r;r.r.s#    r=r.BUILDING_SPHINX_DOCSceZdZdZddZy) _NoDefaultr~cy)Nz r8)r9s r;__repr__z_NoDefault.__repr__sr=N)r?rd)r@rArB no_defaultr8r8r=r;r6r6sJr=r6)named n_objectsn_framesci}d}t|tr|f}|tj}t j r}n!tt r tdfd }|/t|tr|f}nt|ts t|}t j}t|dd} |r||krt|jj}t|D]7\} } | |vs || |vs || s| || <|%t||k\s4|c~S|j }|dz }|r||kr~|S#~wxYw)a@ Retrieve f_locals from all (or the last 'n') stack frames from the calling location. Parameters ---------- of_type Only return objects of this type; can be a single class, tuple of classes, or a callable that returns True/False if the object being tested is considered a match. n_objects If specified, return only the most recent `n` matching objects. n_frames If specified, look at objects in the last `n` stack frames only. named If specified, only return objects matching the given name(s). rNct|Sr7r\)rof_types r; matches_typez'_get_stack_locals..matches_typesc7+ +r=f_backrrrr?rq)rNrdrUmaxsizeinspect isfunctionrrr currentframerrPf_localsrQrHrrA) r?r:r;r<objectsexamined_framesr@ stack_frame local_itemsrrs ` r;_get_stack_localsrLsY.GO%;;'" gz *GnG ,  eS !HEE3'JE&&(K+x6Ko8{3399; *>2::->::>>.1X=>::>>&)-== 5ZZ^^, - 4 5 r=) inject_medianct|tr|g}n|g}td|Ds d}t|t d|D}t d|D}|r|r|ddk7rdg|}g||S)z Transforms raw percentiles into our preferred format, adding the 50th percentile. Raises a ValueError if the percentile sequence is invalid (e.g. outside the range [0, 1]) c3<K|]}d|cxkxrdkncyw)rrNr8rIps r;rKz$parse_percentiles..Zs2Q ! 2sz-`percentiles` must all be in the range [0, 1]c3,K|] }|dks |yw?Nr8rfs r;rKz$parse_percentiles..^sBa!c'B c3,K|] }|dk\s |ywrir8rfs r;rKz$parse_percentiles.._s'Ka!s('Krkrrj)rNfloatr_rsorted) percentilesrcrsub_50_percentilesat_or_above_50_percentiless r;parse_percentilesrrMs+u%"m   2k2 2=oB;BB!''K;'K!K &*DQ*G3*N&)%G,F%G" =  ="< ==r=c<d}tjd|dd|S)z1Escape a string for use in a Polars (Rust) regex.z\\?()|\[\]{}^$#&~.+*-z([z])z\\\1)rr)rre_rust_metacharss r; re_escaperuis(1 66B(),gq 99r=)g0@g(@)show output_path raw_outputfigsizecB|r|S|t|jds&tstsdtj vrdnd} t jddd|zg|j}|rt|j||sytrd d l m}m} | ||Str$d dl} | j#|j%Stj j'ddx} td dl} | j+d 5} | j-|| j/| j1d | j2} t j4| ddddyt7dddd dlm}d dlm}|jA||jCtE|}|jGd|jI||jKy#ttf$rd }t|dwxYw#1swYyxYw)Nz.svgPOLARS_DOT_SVG_VIEWERsvgpngdotz -Nshape=boxz-T)inputzythe graphviz `dot` binary should be on your PATH.(If not installed you can download here: https://graphviz.org/download/)r)SVGdisplay)suffixz%file%T)shell matplotlibrz"should be installed to show graphs) err_prefix err_suffix)ryoff)&rdendswithrrrr`r check_outputencoderFileNotFoundErrorr write_bytesIPython.displayrrrHtmldecoderatempfileNamedTemporaryFilewriteflushrrrunrmatplotlib.imageimagematplotlib.pyplotpyplotfigureimreadr axisimshowrv)r~rvrwrxry output_typegraphrrrrcmdrfilempimgpltimgs r;display_dot_graphrrs   #K(8(A(A&(I >  "bjj 0    )'' M4+#5 6oo>O  [%%e, ~0s5z""  ww%,,.)++::>>"94@ @C M ,,F,; 0t 5! kk(DII6s$/  0  ; )' 7 #ll75>*  3  [ * +) W #D( )2 0 s ,G4AH4HH)qualify_polarsct|tr|j}|j}n,|jj}|jj}|r|dk(s|s|j dr|S|d|S)a Return the module-qualified name of the given object as a string. Parameters ---------- obj The object to get the qualified name for. qualify_polars If False (default), omit the module path for our own (Polars) objects. builtinszpolars.r)rNrGrAr@ __class__ startswith)rrmodulers r;qualified_type_namersq#t||))}}%%  Z 6#4#4Y#? XQtf r=ct|t|s;t|t|s%dt|dt|}t|yy)aB Raise an error if the two arguments are not of the same type. The check will not raise an error if one object is of a subclass of the other. Parameters ---------- current The object the type of which is being checked against. other An object that has to be of the same type. zexpected `other` to be a z, not N)rNrGrr)currentotherrs r;require_same_typers] eT'] +JwU 4T'(;G(D'GH&u-0 2 n 5U +r=ceZdZdZddZy)_NamespaceSuggestMixinzAMixin that adds suggestions to AttributeError on namespace typos.cNddl}tt|Dcgc]}|jdr|}}|j ||dd}|r-dt|j d|d|dd }t |dt|j d|}t |cc}w) Nr_rg333333?)ncutoff'z' object has no attribute z. Did you mean: ?)difflibdirrGrget_close_matchesr@r)r9rrrpublicmatchesrs r; __getattr__z"_NamespaceSuggestMixin.__getattr__s d_FALL4E!FF++D&Ac+J d4j))**DTHL\]def]g\jjklCS!!d4j))**DTHMCS!! Gs B"B"N)rrdr?r)r@rArB__doc__rr8r=r;rrs K "r=rr7)rRz+None | str | Sequence[str] | dict[str, str]r?z2None | str | Sequence[str] | list[tuple[str, str]])rXzobject | Iterator[T]r?zTypeIs[Iterator[T]])rXzIterable[object]r^ztype | tuple[type, ...]r?rq)rXobjectrarqrbrqr?zTypeGuard[Sequence[str | Path]])rXrrbrqr?zTypeGuard[Sequence[bool]])rXrrbrqr?zTypeGuard[Sequence[int]])rXrrbrqr?zTypeGuard[Sequence[Any]])r Sequence[Any]rztype[T]r?zTypeIs[Sequence[T]])rXrrarqrbrqr?zTypeGuard[Sequence[str]]rB)rrr?None)rrdrrrhzPolarsDataType | Noner?z pl.Series)rrr?r)r?rq)rrr?r )rz str | Pathrrqr?rd)rzSequence[str | int]r?ztuple[int, ...])rrr?z list[Any])rz Iterable[str]r?z list[str])rrwrr.r?z int | float)rr,rr.r?r,)rz int | Exprrr.r?zint | float | Expr)r"r+r z dict[str, PolarsDataType | None]r?r+) r?z6type | Collection[type] | Callable[[Any], bool] | Noner:zstr | Collection[str] | Noner;r r<r r?zdict[str, Any]) rzbool | Sequence[bool]rOrwrPrdrQrdr?zSequence[bool])rozSequence[float] | float | Nonercrqr?zSequence[float])rrdr?rd) r~rdrvrqrwzstr | Path | Nonerxrqryztuple[float, float]r?r )rrrrqr?rd)rrrrr?r)x __future__rrDrrrUr collectionsrcollections.abcrrrrr r enumr ior pathlibr typingrrrrrrpolarsrjrrpolars._dependenciesrrrrrepolars._warningsrpolars.datatypesrrrrrrr r!polars.datatypes.groupr"r#r$r%r&r'r(r)r*r+r,polars._typingr-r.rVr1typing_extensionsr2r3r5keysrrQrE__annotations__rrWrrLrSrYr`rnrsrxrzrrrrrrrrrrrrrrr+r,propertyr.getenvr4r6r9r~ NoDefaultrLrRrbrrrurrrrr8r=r;rs" !NN,,   @ 65&7 7"!,#A AI8I w%'WWY RXXZ$HF !H""KF"KK#@D<73 ',E #=A$.,1D D$(DD,1C C$(CC,1 $(',EC C#C=ACC$: ;?    "7   0 IM B= < < 6 6"w=w=;w=w=z T] h !ryy!78 " " J)) * GKF+/  F CF (F F  F  FR     ",KP>/>CG>>8:%)#/ F F F# F  F ! FFR=B8* " "r=