jcUddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZddlZdd lmZdd l m!Z!m"Z"m#Z#dd l m$Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.dd l/m0Z0m1Z1erbddl m2Z2m3Z3m4Z4m5Z5ddlm6Z6m7Z7m8Z8ddlm9Z9m:Z:ddl;mdkrddlm?Z?nddl@m?Z?e6dZAedZBGdde7ZCej>dkrUiDiEiFgZGdeHd<eIdeGDZJ ddd ZKdd$ZLdd)ZMd*d*d+dd0ZNd*d1dd3ZOd*d1dd5ZPd*d1dd7ZQd*d*d+dd9ZRdd<ZSdd>ZT dddFZUddHZVddIZWddJZXddLZYdMdNddRZZddVZ[ddZZ\dd^Z]edddZ^eddfZ^ddiZ^ddnZ_edoZ`GdpdqeaZbejcdrZdGdsdteZeeejfZgeeejfZhdduZiddzZj ddddd{ddZkelfddZmddZnddZod*dddZpddZqdMdd*ddddZrd*dddZsddZtdS)) annotationsN)Counter) Collection GeneratorIterable MappingViewSequenceSized)Enum)BytesIO)Path) TYPE_CHECKINGAnyLiteralTypeVaroverload) functions)_check_for_numpyimport_optional subprocess)numpy)BooleanDateDatetimeDecimalDurationInt64StringTime) FLOAT_DTYPESINTEGER_DTYPES)CallableIteratorMutableMapping Reversible) ParamSpecProtocol TypeGuard) DataFrameExpr)PolarsDataTypeSizeUnit) )TypeIsPTceZdZddZdS)IdentityFunctionfnCallable[P, T]returncdSN)selfr4s Z/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_utils/various.py__call__zIdentityFunction.__call__BsN)r4r5r6r5)__name__ __module__ __qualname__r<r9r=r;r3r3@sHHHHHHr=r3r- zlist[Reversible[Any]]_viewsc#NK|] }tt|V!dSr8)typereversed).0views r; rIHs0"K"KD4#7#7"K"K"K"K"K"Kr= null_values+None | str | Sequence[str] | dict[str, str]r62None | str | Sequence[str] | list[tuple[str, str]]crt|tr!t|S|Sr8) isinstancedictlistitems)rJs r;_process_null_valuesrRKs6+t$$K%%''(((r=valobject | Iterator[T]TypeIs[Iterator[T]]ct|ttfrt|t p9t|tp$t jdkot|tS)NrA)rNrrr rsys version_info_reverse_mapping_views)rSs r; _is_generatorrZTs_ C)X. / / N 38N8N4N U c; ' ' U   ' SJs.^s-22z!V$$222222r=)all)rSr\s `r;_is_iterable_ofrd\s& 2222c222 2 22r=F) allow_strinclude_seriesobjectrerfTypeGuard[Sequence[str | Path]]c|durt|trdSt|r>t|tjr$tj|jtjS|r/t|tj r|jtj kSt|t o0t|tot|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_plSeriesrbytesr rdr rSrerfs r;is_path_or_str_sequencertasEjc22u #  &:c2:#>#>&}SY000 &JsBI66&yBI%% sE " "" . sH % % . C$ - -r=)rfTypeGuard[Sequence[bool]]c4t|r/t|tjr|jtjkS|r/t|t jr|jt jkSt|tot|tS)z;Check whether the given sequence is a sequence of booleans.) rrNrkrlrnbool_rprqrr rdr^rSrfs r;is_bool_sequenceryws'C!t|tjr$tj|jtjS|r3t|tjr|j St|tot|tS)z;Check whether the given sequence is a sequence of integers.) rrNrkrlrmrnintegerrprq is_integerr rdintrxs r;is_int_sequencers&C!t|tjr$tj|jtjS|r/t|tj r|jtj kSt|tot|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) rNrjrrkrlrmrnrorprqrr rdrss r;is_str_sequencersEjc22u #  &:c2:#>#>&}SY000 &JsBI66&yBI%% c8 $ $ Bc)B)BBr=objrc`ddlm}t||o|jS)z9Indicate if the given object is a basic/unaliased column.r)r*) polars.exprr*rNmeta is_column)rr*s r;rrs7      c4 9SX%7%7%9%99r=Nonec^|*tjdttdSdS)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 UserWarningfind_stacklevelrs r;warn_null_comparisonrs@ { k &((      {r=namerjrngrangernPolarsDataType | None pl.Seriesc,|pt}|r)tj|j|j|j|d}n:tj|j|j|jd|}||S)z/Fast conversion of the given range to a Series.T)startendsteprneager)rrrr) rr}F int_rangerstoprcastalias)rrrnrs r;range_to_seriesrs NUE  )T    )   $u++  ;;t  r=slicecBt|j|j|jS)z.Return the given range as an equivalent slice.)rrrr)rs r;range_to_slicers CHch / //r=cv ddlm}d|jvrdSn#t$rYdSt$rYdSwxYwdS)Nr get_ipython IPKernelAppFT)IPythonrconfig ImportErrorAttributeErrorrs r; _in_notebookrsw''''''  4 4 45 5 uu uu 4s 6 66cV ddl}|S#t$rYdSwxYw)NrF)marimorunning_in_notebookr)mos r;_in_marimo_notebookrsG%%''' uus  (( int | Nonec t|ttfrdnt|S#t$rYdSwxYw)zLReturn length of (non-string/dict) sequence; returns None for non-sequences.N)rNrjrOlen TypeErrorrs r;arrlenrsL!#T{33AttSA tts ,/ ==T)check_not_directorypath str | Pathrctj|}|rStj|r4tj|rd|d}t ||S)z>Create a string path, expanding the home directory if present.zexpected a file path; z is a directory)osr expanduserexistsisdirIsADirectoryError)rrmsgs r;normalize_filepathrst 7  d # #D% GNN4 % GMM$  % ?t>>>$$$ Kr=versionSequence[str | int]tuple[int, ...]ct|tr|d}td|DS)zASimple version parser; split into a tuple of ints for comparison..c 3vK|]4}ttjddt|V5dS)z\DN)r~resubrj)rGvs r;rIz parse_version..s>AAARVE2s1vv..//AAAAAAr=)rNrjsplittuple)rs r; parse_versionrsB'3%--$$ AAAAA A AAr=values Sequence[Any] list[Any]cNtjfd|DS)zMReturn unique list of sequence values, maintaining their order of appearance.c2g|]}|v||Sr9r9)rGradd_seens r; z"ordered_unique..s* < < brr>kb kilobytesi>mb megabytesi>gb gigabytesi@>tb terabyteslz9`unit` must be one of {'b', 'kb', 'mb', 'gb', 'tb'}, got ) ValueError)rrrs r;rrs ~ $ $ $Dy $ $ $G| $ $ $G| $ $ $G|TDTToor=dfr)schema dict[str, PolarsDataType | None]c  |s;|jD]!}|tkrd|}t |"hd}d}dddd||dz|dzdz|dzd z|dzd zd zd d-fd }i}|D]\}}||tkr t|j} tj |j dd} tj | j dk| dz| dzj ddj | d||<t'|dd(||j|j||<3|t.kr2tj |j |d||<p|t0krtj tj |j dktj |dztj |dzj ddj |d||<C|t2krftj ||t6t3d |||<|t:kr/tj |d!d"d#||<|t>vrtj |j d$d} tCj | j dk| |||<~|tDvs|tFkr:tj |j d%d&} tj |j d%d'} tCj tj |j $d(tj |j %&|ztCj tj |j dktj | tj |j $d)tj |j d*d+tCj'| j d$d| d+,t|||<||j|kr*tj ||||<|r |j(d.i|n|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µsmssmhdwtd str | Noner6rch|dn-tfdtjd|DS)Nc3rK|]1\}}t||zV2dSr8)r~strip)rGvaluerduration_scalings r;rIzH_cast_repr_strings_with_schema..str_duration_..ZsTE4E -djjll;;r=z([+-]?\d+)(\D+))sumrfindall)r rs r; str_duration_z5_cast_repr_strings_with_schema..str_duration_VsVz D#%:.@"#E#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 rr6rr9))is_emptyrrrrrQ base_typer time_unitrcolrjreplacewhen len_bytesthen otherwiserstrptimegetattrdtreplace_time_zonerrrr map_elementsrrrreplace_strictr! replace_allrpr rcontains to_lowercaseis_in concat_str with_columns)rrtprspecial_floatsns_secr cast_colsctp_baser  int_string integer_partfractional_partrs @r;_cast_repr_strings_with_schemar?,s& ;;==%)""$$ % %BV||UrUUnn$433NF  b[ b[2  e^b e^b 1 $        I@1@12 >||~~))"2<00E!HHL((b99F15??,,233T!l*++Yq;//q"'+BCC ! 2{D11=#,Q??DDQU1XXNNOOT!%((,//7788T!%((,..|SAABBY (,88RHH+&)T&\\T"XX'! *ry|## uQxx}}R00 ! +4 3  ,   KKK s &)??N)rCrrDrEr6r@)r>r?r@rHr9r=r;rBrBs(      r=rBBUILDING_SPHINX_DOCSceZdZdZddZdS) _NoDefault NO_DEFAULTr6rjcdS)Nz r9)r:s r;__repr__z_NoDefault.__repr__s~r=N)r6rj)r>r?r@ no_defaultrNr9r=r;rKrKs/Jr=rKchtttjj}t j}d} |rft j|}||s-t|j ddx}r"|dr |j }|dz }nn|f~n#~wxYw|S)z Find the first place in the stack that is not inside Polars. Taken from: https://github.com/pandas-dev/pandas/blob/ab89c53f48df67709a533b6a95ce3d911871a0a8/pandas/util/_exceptions.py#L30-L51 r co_qualnameNzsingledispatch.r) rjr rp__file__parentinspect currentframegetfile startswithr+f_codef_back)pkg_dirframenfnamequalnames r;rrs$r{##*++G  " "E A OE**E(( $U\=$GGG ''(9::   Q  EE Hs A(B,,B/messagecategory type[Warning]kwargsc Htjd||td|dS)a  Issue a warning. Parameters ---------- message The message associated with the warning. category The warning category. **kwargs Additional arguments for `warnings.warn`. Note that the `stacklevel` is determined automatically. )r_r`rNr9rrr)r_r`rbs r; issue_warningresA M(7H7HLRr=)named n_objectsn_framesof_type6type | Collection[type] | Callable[[Any], bool] | Nonerfstr | Collection[str] | Nonergrhdict[str, Any]ci}d}t|tr|f}| tj}t jr}n*tt rtd fd }|=t|tr|f}n$t|tst|}t j }t|dd} |r||krt|j }t|D];\} } | |vr2|| |vr, || r| || <|t||kr|c~S<|j}|d z }|r||k~n#~wxYw|S) 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). rNrrr6r^c$t|Sr8ra)rris r; matches_typez'_get_stack_locals..matches_type%sc7++ +r=rYrrrr6r^)rNrjrWmaxsizerT isfunctionrrrrUr+rPf_localsrQrFrrY) rirfrgrhobjectsexamined_framesro stack_frame local_itemsrrs ` r;_get_stack_localsrxs.GO%;'"", gz * * %GnnG , , , , , ,  eS ! ! HEEE3'' JJE&((K+x66K !o88{399;;< *>2:->:>>.11X=>:>>&))-== 5Z^^, - - 4 5 5r=) inject_median percentilesSequence[float] | float | NonerSequence[float]c*t|tr|g}n|g}td|Dsd}t|t d|D}t d|D}|r|r |ddkrdg|}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]) Nc36K|]}d|cxkodkncVdS)rrNr9rGps r;rIz$parse_percentiles..s622Q ! 222222r=z-`percentiles` must all be in the range [0, 1]c3&K|] }|dk|V dS?Nr9rs r;rIz$parse_percentiles..s&BBa!c''''''BBr=c3&K|] }|dk|V dSrr9rs r;rIz$parse_percentiles..s&'K'Ka!s(((((('K'Kr=rr)rNfloatrcrsorted)rrrsub_50_percentilesat_or_above_50_percentiless r;parse_percentilesrxs+u%%"m   22k222 2 2=ooBB;BBBBB!''K'K;'K'K'K!K!KH &H*DQ*G3*N*N&)%G,F%G" =  ="< ==r=rc: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_escapers+1 6,(,,,gq 9 99r=)g0@g(@)show output_path raw_outputfigsizedotrrstr | Path | Nonerrtuple[float, float]rc|r|S|"t|ds*tstsdtjvrdnd} t jddd|zg|}n'#ttf$rd }t|dwxYw|r"t| ||sdStrd d l m}m} | ||Str,d dl} | |Stjddx} d dl} | d 5} | || | d | j} t j| ddddn #1swxYwYdSt7dddd dlm}d dlm}| ||!tE|}|#d|$||%dS)Nz.svgPOLARS_DOT_SVG_VIEWERsvgpngrz -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)roff)&rjendswithrrrrr check_outputencoderFileNotFoundErrorr write_bytesIPython.displayrrrHtmldecodertempfileNamedTemporaryFilewriteflushr%rrunrmatplotlib.imageimagematplotlib.pyplotpyplotfigureimreadr axisimshowr)rrrrr output_typegraphrrrrcmdrfilempimgpltimgs r;display_dot_graphrs   #K(8(8(A(A&(I(I # >> $   $ #bj 0 0    )' M4+#5 6hoo>O>O    * +))) W #D( )- [%%e,,, t~~ 00000000wss5zz"""   ww%,,..*+++:>>"94@@ @C M OOO,,F,;; 0t 5!!! kk(DI66s$////  0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 4 ;    )((((('''''' 7 ###ll75>>**  3  ts/B$B,7AGG"%G")qualify_polarsrct|tr|j}|j}n|jj}|jj}|r|dks|s|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)rNrEr?r> __class__rW)rrmodulers r;qualified_type_namers#t&|)}%  Z   #)#4#4Y#?#?    t  r=currentotherct|t|sNt|t|s3dt|dt|}t|dSdS)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)rNrErr)rrrs r;require_same_typers eT']] + +JwU 4T4T 2(;G(D(D 2 2&u-- 2 2 nn r=r8)rJrKr6rL)rSrTr6rU)rSr[r\r]r6r^)rSrgrer^rfr^r6rh)rSrgrfr^r6ru)rSrgrfr^r6rz)rSrgrfr^r6r)rSrgrer^rfr^r6rrp)rrr6r)rrjrrrnrr6r)rrr6r)r6r^)rrr6r)rrrr^r6rj)rrr6r)rrr6r)rrr6r)rr~rr,r6r)rr*rr,r6r*)rrrr,r6r)rr)rrr6r))r6r~)r_rjr`rarbrr6r) rirjrfrkrgrrhrr6rl)rrjr`rar6r) rr{r|r~r}rjr~rjr6r)rrrr^r6r)rrjr6rj) rrjrr^rrrr^rrr6r)rrrr^r6rj)rrrrr6r)u __future__rrTrrrWr collectionsrcollections.abcrrrrr r enumr ior pathlibr typingrrrrrpolarsrprrpolars._dependenciesrrrrrkpolars.datatypesrrrrrrrrpolars.datatypes.groupr r!r"r#r$r%r&r'r(r)r*polars._typingr+r,rXr/typing_extensionsr0r1r3keysrrQrC__annotations__rrYrRrZrdrtryrrrrrrrrrrrrrrrr?r@propertyrBgetenvrIrKrOrL NoDefaultrrerxrrzrrrrrrrr9r=r;rs""""""" !!!!!!NNNNNNNNNN,,,,,,                    @???????I 6555555555&&&&&&&&77777777 7""!!!!!!!,,,,,, #A AIIIII8III w%'WWYY RXXZZ$HFHHHH"U"K"KF"K"K"KKK@D3333 ',E.,1DDDDDD,1CCCCCC,1',ECCCCCC$::::    ;?      0000    IM      BBBB==== <<< < 666 6"w=w=w=w=z WT]]     h   !ry!788 " J) *     B(GKF+/  FFFFFFT6A",KP>>>>>>8::::%)#/ FFFFFFR=B8r=