hb%SSKJr SSKrSSKrSSKrSSKrSSKrSSKJr SSK J r J r J r J r JrJr SSKJr SSKJr SSKJr SSKJrJrJrJrJrJr SSKrSS KJr SS K!J"r"J#r#J$r$ SS K!J%r& SS K'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/ SS K0J1r1J2r2 \(ahSSK J3r3J4r4J5r5 SSKJ6r6J7r7 SSK8J9r9J:r: \RvS:aSSKJr>J?r? OSSK=J>r>J?r? \>"S5r@\"S5rA\RvS:aC0R50R50R5/rES\FS'\G"S\E55rHSMSNSjjrISOSjrJSPSjrKSSS.SQS jjrLSS!.SRS"jjrMSS!.SSS#jjrNSS!.STS$jjrOSSS.SUS%jjrPSVS&jrQSWS'jrRSMSXS(jjrSSYS)jrTSZS*jrUSZS+jrVS[S,jrWS-S..S\S/jjrXS]S0jrYS^S1jrZS_S2jr[\S`S3j5r\\SaS4j5r\SbS5jr\ScS6jr]\"S75r^"S8S9\_5r`\R"S:5rb"S;S<\5rc\cRrd\\cRreSdS=jrfSeS>jrgSMSSSS?.SfS@jjjrh\i4SgSAjjrjShSBjrkSZSCjrlSSD.SiSEjjrmSjSFjrnS-SSSGSH.SkSIjjroSSJ.SlSKjjrpSmSLjrqg)n) annotationsN)Counter) Collection GeneratorIterable MappingViewSequenceSized)Enum)BytesIO)Path) TYPE_CHECKINGAnyCallableLiteralTypeVaroverload) functions)_check_for_numpyimport_optional subprocess)numpy)BooleanDateDatetimeDecimalDurationInt64StringTime) FLOAT_DTYPESINTEGER_DTYPES)IteratorMutableMapping Reversible) DataFrameExpr)PolarsDataTypeSizeUnit) )TypeIs)r* ) ParamSpec TypeGuardPTr* zlist[Reversible[Any]]_viewsc#J# UHn[[U55v M g7fN)typereversed).0views O/home/james-whalen/.local/lib/python3.13/site-packages/polars/_utils/various.py r<Cs"KFD4#7#7Fs!#cb[U[5(a[UR55$U$r6) isinstancedictlistitems) null_valuess r;_process_null_valuesrCFs++t$$K%%'((c[U[[45=(a [U[5(+=(dA [U[5=(d* [ R S:=(a [U[5$)Nr2)r>rrr rsys version_info_reverse_mapping_views)vals r; _is_generatorrJOsV C)X. / N 38N4N U c; ' U    ' SJs# UHn[UT5v M g7fr6r>)r9xeltypes r;r<"_is_iterable_of..Ys2cz!V$$cs)all)rIrOs `r;_is_iterable_ofrRWs 2c2 22rDF) allow_strinclude_seriesc USLa[U[5(ag[U5(aN[U[R5(a/[R "UR [R5$U(a<[U[R5(aUR [R:H$[U[5(+=(a- [U[5=(a [U[[45$)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)r>strrnpndarray issubdtypedtypestr_plSeriesrbytesr rRr rIrSrTs r;is_path_or_str_sequencer`\sEjc22 #  :c2::#>#>}}SYY00 JsBII66yyBII%% sE "" . sH % . C$ -rD)rTcn[U5(a<[U[R5(aUR[R :H$U(a<[U[ R5(aUR[ R:H$[U[5=(a [U[5$)z;Check whether the given sequence is a sequence of booleans.) rr>rWrXrZbool_r\r]rr rRboolrIrTs r;is_bool_sequencerersqC!rWrXrYrZintegerr\r] is_integerr rRintrds r;is_int_sequencerj}swC!rWrXr\r]r rVrds r; is_sequencerlsO S ! Ajbjj&A 3H-XN %3$ $rDcUSLa[U[5(ag[U5(aN[U[R5(a/[R "UR [R5$U(a<[U[R5(aUR [R:H$[U[5=(a [U[5$)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) r>rVrrWrXrYrZr[r\r]rr rRr_s r;is_str_sequencernsEjc22 #  :c2::#>#>}}SYY00 JsBII66yyBII%% c8 $ Bc)BBrDcfSSKJn [X5=(a URR 5$)z9Indicate if the given object is a basic/unaliased column.r)r') polars.exprr'r>meta is_column)objr's r;rrrrs" c 9SXX%7%7%99rDcPUc#[R"S[[5S9 gg)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_stacklevelrss r;warn_null_comparisonr{s$ { k &( rDcbU=(d [nUR5(a7[R"URUR UR USS9nOD[R"URUR UR SS9RU5nURU5$)z/Fast conversion of the given range to a Series.T)startendsteprZeager)r}r~rr) rrhF int_ranger}stoprcastalias)namerngrZranges r;range_to_seriesrs NUE  ))T  )) $u+  ;;t rDcX[URURUR5$)z.Return the given range as an equivalent slice.)slicer}rr)rs r;range_to_slicers CHHchh //rDcvSSKJn SU"5R;agg![a g[a gf=f)Nr get_ipython IPKernelAppFT)IPythonrconfig ImportErrorAttributeErrorrs r; _in_notebookrsF'  4 4 4 5   s 8 88cLSSKnUR5$![a gf=f)NrF)marimorunning_in_notebookr)mos r;_in_marimo_notebookrs+%%'' s  ##ct[U[[45(aS$[U5$![a gf=f)zLReturn length of (non-string/dict) sequence; returns None for non-sequences.N)r>rVr?len TypeErrorrzs r;arrlenrs8!#T{33tASA s* * 77T)check_not_directoryc[RRU5nU(aZ[RRU5(a6[RR U5(aSU<S3n[ U5eU$)z>Create a string path, expanding the home directory if present.zexpected a file path; z is a directory)ospath expanduserexistsisdirIsADirectoryError)rrmsgs r;normalize_filepathrs_ 77  d #D GGNN4 GGMM$  &tho>$$ KrDcr[U[5(aURS5n[SU55$)zASimple version parser; split into a tuple of ints for comparison..c 3v# UH/n[[R"SS[U555v M1 g7f)z\DN)riresubrV)r9vs r;r< parse_version..s)AARVVE2s1v.//s79)r>rVsplittuple)versions r; parse_versionrs.'3--$ AA AArDc[5nURnUVs/sHo3U;aM U"U5(aMUPM sn$s snf)zMReturn unique list of sequence values, maintaining their order of appearance.)setadd)valuesseenadd_rs r;ordered_uniquers6UD 88D br^>kb kilobytesi>mb megabytesi>gb gigabytesi@>tb terabyteslz9`unit` must be one of {'b', 'kb', 'mb', 'gb', 'tb'}, got ) ValueError)rrrs r;rrsq ~ $ $Dy $ $G| $ $G| $ $G|KD8TorDc ` ^UR5(d<URR5HnU[:wdMSU<3n[ U5e 1SknSnSSSSUUS-US-S-US-S -US-S -S -S . mS)U4S jjn0nUR 5GHupUcM UR 5[:XGa [UR5n [R"U5RRSS5n [R"U RR5S:H5RU S-5R!U S-5RR#SS5RR%U S5Xx'['USS 5b,XxR(R+UR,5Xx'GM.GM1U[.:Xa5[R"U5RR%US5Xx'GMpU[0:Xa[R"[R"U5RR5S:H5R[R"U5S-5R![R"U5S-5RR#SS5RR%US5Xx'GMDU[2:XaT[R"U5R5U[6S9R9[3S55R9U5Xx'GMU[::Xa-[R"U5R=SSS.5Xx'GMU[>;a[R"U5RRAS S5n [BR"U RR5S:5RU 5R9U5Xx'GMgU[D;dUR 5[F:XGa3[R"U5RRS!S"5n [R"U5RRS!S#5n [BR"[R"U5RRIS$5[R"U5RRK5RMU5-5R[BR"[R"U5RR5S:5R[R"U555R[R"U5RRIS%55R[R"U5RRS&S'55R![BRN"U RRAS S5U S'S(95R9[5R9U5Xx'GMX RU:wdGM[R"U5R9U5Xx'GM U(aURP"S*0UD6$U$)+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 >infnan+inf-infiʚ;rii@B<i) nsusuµsmssmhdwcb>UcS$[U4Sj[R"SU555$)Nc3f># UH&up[U5TUR5-v M( g7fr6)ristrip)r9valuerduration_scalings r;r<H_cast_repr_strings_with_schema..str_duration_..Us/#EKEE -djjl;;#Es.1z([+-]?\d+)(\D+))sumrfindall)tdrs r; str_duration_5_cast_repr_strings_with_schema..str_duration_Qs<z  #%::.@"#E rDNz[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 | Nonereturn int | Noner))is_emptyschemarrrrA base_typer time_unitrcolrVreplacewhen len_bytesthen otherwiserstrptimegetattrdtreplace_time_zonerrr r map_elementsrrrreplace_strictr" replace_allr\r!rcontains to_lowercaseis_in concat_str with_columns)dfrtprspecial_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 >||~)"2<<0EE!HLL((b9FF155??,23T!l*+Yq;/Sq"S'+BC  2{D1=#,t uuQx||44RD tFF1558<<113q89T!%%(\12YquuQx+56Sq"S"n5  xEE!H!\-e\DT(4.)T"X  w uuQx66u7UV ~%UU1X\\55j"E GGJNN446:;@@LQQRTU |#r||~'@ uuQx||334DdK "#%%(,,"6"67G"NGGa --m<%%(,,335;;NKLT"''!%%(,,"8"8":Q">?DDQUU1XNOT!%%(,,//78T!%%(,,..z3?@Y (,,88RH+&)T&\T"X' *yy|# uuQx}}R0 A D,52?? 'Y '<"rr)selfinstanceclss r;__get__sphinx_accessor.__get__sM  99&x55 ;>  , K s ))>>rN)r$rr%ztype[NS]rr)__name__ __module__ __qualname____firstlineno__r&__static_attributes__rrDr;r r s#    rDr BUILDING_SPHINX_DOCSc"\rSrSrSrSSjrSrg) _NoDefaulti NO_DEFAULTcg)Nz r)r#s r;__repr___NoDefault.__repr__srDrN)rrV)r(r)r*r+ no_defaultr2r,rrDr;r/r/sJrDr/c[[[R5R5n[ R "5nSnU(a|[ R"U5nURU5(d4[URSS5=n(a(URS5(aURnUS- nOO U(aM|AU$!Af=f)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) rVr r\__file__parentinspect currentframegetfile startswithrf_codef_back)pkg_dirframenfnamequalnames r;ryrys$r{{#**+G  "E AOOE*E(($U\\=$GGG''(9:: Qe  H s BC Cc F[R"SX[5S.UD6 g)z 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. )messagecategoryruNrrvrwry)rErFkwargss r; issue_warningrIs% MM7HLRrD)named n_objectsn_framesc^0nSn[U[5(aU4nUc[Rn[R "T5(aTnO([T[ 5(a [T5mSU4SjjnUb9[U[5(aU4nO [U[5(d [U5n[R"5n[USS5nU(aXS:a[URR55n[U5HEupX;dM UbX;dMTbU"U 5(dM(XU 'UcM1[U5U:dMBUs A$ UR nUS- nU(aXS:aMAU$!Af=f)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). rNc>[UT5$r6rM)rsof_types r; matches_type'_get_stack_locals..matches_type sc7+ +rDr>rrsrrrc)r>rVrFmaxsizer9 isfunctionrrrr:rr@f_localsrAr8rr>) rOrJrKrLobjectsexamined_framesrP stack_frame local_itemsrrss ` r;_get_stack_localsrZsZ.GO%;;'"" gz * *GnG ,  eS ! !HEE3''JE&&(K+x6Ko8{3399;c@[R"UU[5S9 g)N)rFrurG)rrFs r; _polars_warnr\Es MM "$rDc [U[5(aU/U-OUnU[U5:wa#SUS[U5SUSUS3 n[U5eU$)zAEnsure the given bool or sequence of bools is the correct length.zthe length of `z` (z ) does not match the length of `))r>rcrr)rn_match value_name match_namerrs r; extend_boolrbMsi#-UD"9"9eWw uF#f+j\S[M:--7LG9A G o MrDc[[RS5(a[RR5=(a [RS:g=(d S[ R ;=(de S[ R ;=(dK [ R RS5S:H=(d" [ R RS5S:H=(d" [ R RS 5S :H$g ) z Determine (within reason) if we are in an interactive terminal that supports color. Note: this is not exhaustive, but it covers a lot (most?) of the common cases. isattywin32ANSICON WT_SESSION TERM_PROGRAMvscodeTERMzxterm-256colorPYCHARM_HOSTED1F)hasattrrFstdoutrdplatformrenvirongetrrDr; in_terminal_that_supports_colourrr^s szz8$$ JJ     '> *>2::->::>>.1X=>::>>&)-== 5ZZ^^, - 4 5 rD) inject_medianc[U[5(aU/nOUc/n[SU55(d Sn[U5e[ SU55n[ SU55nU(aU(a USS:waS/UQn/UQUQ$)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]) c3L# UHnSUs=:*=(a S:*Os v M g7f)rrNrr9ps r;r<$parse_percentiles..s2kQ ! ks"$z-`percentiles` must all be in the range [0, 1]c34# UHoS:dM Uv M g7f?Nrrvs r;r<rxsB;ac';  c34# UHoS:dM Uv M g7frzrrvs r;r<rxs'K;as(;r|rr{)r>floatrQrsorted) percentilesrsrsub_50_percentilesat_or_above_50_percentiless r;parse_percentilesrss+u%%"m   2k2 2 2=oB;BB!''K;'K!K &*DQ*G3*N&)%G,F%G" =  ="< ==rDc>Sn[R"SUS3SU5$)z1Escape a string for use in a Polars (Rust) regex.z\\?()|\[\]{}^$#&~.+*-z([z])z\\\1)rr)rre_rust_metacharss r; re_escapers(1 66B(),gq 99rD)g0@g(@)show output_path raw_outputfigsizecHU(aU$[5(d#[5(dS[R;aSOSn[R "SSSU-/UR 5S9nU(a[U5RU5 U(dg[5(aS S K J nJ n U "U"U55$[5(a$S SKn U RUR!55$[RR#SS5=n boS SKn U R'S S 9n U R)U5 U R+5 U R-S U R.5n [R0"U SS9 SSS5 g[3SSSS9 S SKJn S SKJn UR=US9 UR?[AU55nURCS5 UREU5 URG5 g![[4a Sn[U5Sef=f!,(df  g=f)NPOLARS_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)SVGdisplayz.svg)suffixz%file%T)shell matplotlibrz"should be installed to show graphs) err_prefix err_suffix)roff)$rrrrpr check_outputencoderFileNotFoundErrorr write_bytesIPython.displayrrrHtmldecoderqtempfileNamedTemporaryFilewriteflushrrrunrmatplotlib.imageimagematplotlib.pyplotpyplotfigureimreadr axisimshowr)rrrrr output_typegraphrrrrcmdrfilempimgpltimgs r;display_dot_graphrs  >>  "bjj 0   )'' M4+#5 6oo>O  [%%e, ~~0s5z""   ww%,,.)++::>>"94@ @C M ,,F,;t 5! kk(DII6s$/ <  ; )' 7 #ll75>*  3  [ * +) W #D( )2<; s*G1"AH1H H!)qualify_polarsc[U[5(aURnURnO,URRnURRnU(a#US:XdU(dUR S5(aU$USU3$)z 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)r>r7r)r( __class__r<)rsrmodulers r;qualified_type_namersw#t||))}}%%  Z 6#4#4Y#?#? XQtf rDc[U[U55(dB[U[U55(d'S[U5<S[U5<3n[U5egg)a 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)r>r7rr)currentotherrs r;require_same_typersc eT'] + +JwU 4T4T'(;G(D'GH&u-0 2 n 5U +rDr6)rBz+None | str | Sequence[str] | dict[str, str]rz2None | str | Sequence[str] | list[tuple[str, str]])rIzobject | Iterator[T]rzTypeIs[Iterator[T]])rIzIterable[object]rOztype | tuple[type, ...]rrc)rIobjectrSrcrTrcrzTypeGuard[Sequence[str | Path]])rIrrTrcrzTypeGuard[Sequence[bool]])rIrrTrcrzTypeGuard[Sequence[int]])rIrrTrcrzTypeGuard[Sequence[Any]])rIrrSrcrTrcrzTypeGuard[Sequence[str]]rR)rsrrNone)rrVrrrZzPolarsDataType | Nonerz pl.Series)rrrr)rrc)rsrrr)rz str | PathrrcrrV)rzSequence[str | int]rztuple[int, ...])rz Sequence[Any]rz list[Any])rz Iterable[str]rz list[str])rrirr)rz int | float)rr'rr)rr')rz int | Exprrr)rzint | float | Expr)rr&rz dict[str, PolarsDataType | None]rr&)rri)rErVrF type[Warning]rHrrr) rOz6type | Collection[type] | Callable[[Any], bool] | NonerJzstr | Collection[str] | NonerKrrLrrzdict[str, Any])rrVrFrrr) rzbool | Sequence[bool]r_rir`rVrarVrzSequence[bool])rzSequence[float] | float | NonersrcrzSequence[float])rrVrrV) rrVrrcrzstr | Path | Nonerrcrztuple[float, float]rr)rsrrrcrrV)rrrrrr)r __future__rr9rrrFrv collectionsrcollections.abcrrrrr r enumr ior pathlibr typingrrrrrrpolarsr\rrpolars._dependenciesrrrrrWpolars.datatypesrrrrrrrr polars.datatypes.groupr!r"r#r$r%r&r'polars._typingr(r)rGr,typing_extensionsr.r/r0r1keysrrAr4__annotations__rrHrCrJrRr`rerjrlrnrrr{rrrrrrrrrrrrpropertyr getenvr-r/r4 NoDefaultryrIrZrxr\rbrrrrrrrrrDr;rs" !NN,   @DD&7 7"!, 7"//:#A Aw%'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 yy!78 " " J)) *  B(GKF+/  F CF (F F  F  FT6A     ",KP>/>CG>>8:%)#/ E E E# E  E ! EEP=B8rD