g| UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl mZddlmZddlmZddlmZmZmZmZmZmZmZddlZddlmcmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'ejPd k\r dd lm)Z)m*Z*m+Z+n dd l,m)Z)m*Z*m+Z+ejPdk\r ddlm-Z-m.Z.nddl,m-Z-m.Z.erddlZ/ddl0Z1ddl"m2Z2ddl3m4Z5ddl6m7Z8ede#Z9edZ:e.dZ;edZ<e*dedee*dde:eZ[e dU dVd?Z\e dWd@Z\ dX dYdAZ\dZd[dBZ]edCZ^dDeQdE< GdFdDZ_dGZ` d\dHZad]dIZbycc}}w)^zUtility routines.) annotationsN)IteratorMappingMutableMapping)deepcopy)groupby) itemgetter) TYPE_CHECKINGAnyCallableLiteralTypeVarcastoverload)is_pandas_dataframeis_polars_dataframe) IntoDataFrame) SchemaBase SchemaLike Undefined) )Protocol TypeAliasTyperuntime_checkable)r ) Concatenate ParamSpec)IntoExpr) DataFrame)StandardType_TTIntoDataFrame)boundTPR WrapsFunc.) type_params WrappedFunc WrapsMethodz Callable[Concatenate[T, ...], R] WrappedMethodc"eZdZ d ddZy) DataFrameLikecyN)self nan_as_null allow_copys F/var/www/openai/venv/lib/python3.12/site-packages/altair/utils/core.py __dataframe__zDataFrameLike.__dataframe__>sN)FT)r2boolr3r7return DfiDataFrame)__name__ __module__ __qualname__r5r0r6r4r-r-<s%=A59 r6r-ONQG)ordinalnominal quantitativetemporalgeojson)argmaxargminaveragecountdistinctmaxmeanmedianminmissingproductq1q3ci0ci1stderrstdevstdevpsumvalidvaluesvariance variancep exponential exponentialb) row_numberrank dense_rank percent_rank cume_distntilelaglead first_value last_value nth_value)Qyearquartermonthweekday dayofyeardatehoursminutesseconds milliseconds yearquarteryearquartermonth yearmonth yearmonthdateyearmonthdatehoursyearmonthdatehoursminutes yearmonthdatehoursminutessecondsyearweek yearweekdayyearweekdayhoursyearweekdayhoursminutesyearweekdayhoursminutesseconds yeardayofyear quartermonth monthdatemonthdatehoursmonthdatehoursminutesmonthdatehoursminutessecondsweekday weeksdayhours weekdayhoursweekdayhoursminutesweekdayhoursminutessecondsdayhoursdayhoursminutesdayhoursminutesseconds hoursminuteshoursminutessecondsminutessecondssecondsmillisecondsutcyear utcquarterutcmonthutcweekutcday utcdayofyearutcdateutchours utcminutes utcsecondsutcmillisecondsutcyearquarterutcyearquartermonth utcyearmonthutcyearmonthdateutcyearmonthdatehoursutcyearmonthdatehoursminutes#utcyearmonthdatehoursminutesseconds utcyearweekutcyearweekdayutcyearweekdayhoursutcyearweekdayhoursminutes!utcyearweekdayhoursminutessecondsutcyeardayofyearutcquartermonth utcmonthdateutcmonthdatehoursutcmonthdatehoursminutesutcmonthdatehoursminutesseconds utcweekdayutcweekdayhoursutcweekdayhoursminutesutcweekdayhoursminutesseconds utcdayhoursutcdayhoursminutesutcdayhoursminutessecondsutchoursminutesutchoursminutessecondsutcminutessecondsutcsecondsmillisecondsz (?P.*)z (?P{})|z(?Pcount)z (?Pcount)z(?P{})z (?P{})z(?P{}))fieldtype agg_countop_count aggregate window_optimeUnit)rrrrz mixed-integermixed-integer-floatcomplexintegerfloatingrC categoricalcatrA>bytesmixedstringbooleanunicoderrB>rptimeperioddatetime timedelta datetime64 timedelta64rDz.I don't know how to infer vegalite type from 'z'. Defaulting to nominal. stacklevel) pandas.api.typesrhasattrrordered categoriestolistwarningswarn)datartyps r4infer_vegalite_type_for_pandasrs- d5 )C  '$"6488;K;K488..55788 Q Q   .to_list_if_arrayxs" c2:: &::< Jr6rcategoryrr7r)r timestampc"|jSr/) isoformat)xs r4z+sanitize_pandas_dataframe..s Q[[]r6NaTrField " " has type "^" which is not supported by Altair. Please convert to either a timestamp or a numerical value.r> Int8Int16Int32Int64UInt8UInt16UInt32UInt64Float32Float64)rpandascopyrcolumns RangeIndexastyper ValueErrorindex MultiIndexdtypesitemsrobjectwherenotnullrapplyreplacer rrisnullisinf) dfpdcol_namemsgr dtype_itemr  dtype_namecol bad_valuesr s @r4sanitize_pandas_dataframerAMsK( B"**bmm,ZZ&&s+ JJ(C(:8,G// S/ ! "((BMM*2o"**bmm,2o iioo' Z]+1 Z  #X,%%f-C99S[[]D9BxL 8 #X,%%f-C99S[[]D9BxL 6 !h<..v6BxL 9 $X,%%f-C99S[[]D9BxL  " "#< =8 ""#:;CCE2N xL " "; /(<w7  S/ !  " ": .    X,%%f-C99S[[]D9BxL eRZZ 0h<..v6BxL eR[[ 1X,C 5J::f-33ZKFBxL f_X,%%f-334DEC99S[[]D9BxL_(` Ir6cF|j}g}d}t|j}|jD]T\}}|tj k(r]|r[|j t j|jtjjj|w|tj k(r>|j t j|jj||tjk(rB|j t j|jj|d|tjk(rd|d|d}t||j |W|j|S)z3Sanitize narwhals.DataFrame for JSON serialization.z%Y-%m-%dT%H:%M:%Sz%.frrr)schemar to_nativer1nwDateappendr?rDatetimedt to_stringDurationr-select)rrCr*local_iso_fmt_string is_polarsnamer r<s r4sanitize_narwhals_dataframerPs@[[F G.#DNN$45I||~ e BGG   NNt !!"++.11;;>> import pandas as pd >>> data = pd.DataFrame({"foo": ["A", "B", "A", "B"], "bar": [1, 2, 3, 4]}) >>> parse_shorthand("name") == {"field": "name"} True >>> parse_shorthand("name:Q") == {"field": "name", "type": "quantitative"} True >>> parse_shorthand("average(col)") == {"aggregate": "average", "field": "col"} True >>> parse_shorthand("foo:O") == {"field": "foo", "type": "ordinal"} True >>> parse_shorthand("min(foo):Q") == { ... "aggregate": "min", ... "field": "foo", ... "type": "quantitative", ... } True >>> parse_shorthand("month(col)") == { ... "field": "col", ... "timeUnit": "month", ... "type": "temporal", ... } True >>> parse_shorthand("year(col):O") == { ... "field": "col", ... "timeUnit": "year", ... "type": "ordinal", ... } True >>> parse_shorthand("foo", data) == {"field": "foo", "type": "nominal"} True >>> parse_shorthand("bar", data) == {"field": "bar", "type": "quantitative"} True >>> parse_shorthand("bar:O", data) == {"field": "bar", "type": "ordinal"} True >>> parse_shorthand("sum(bar)", data) == { ... "aggregate": "sum", ... "field": "bar", ... "type": "quantitative", ... } True >>> parse_shorthand("count()", data) == { ... "aggregate": "count", ... "type": "quantitative", ... } True r) is_data_typez{agg_count}\(\)z{aggregate}\({field}\)z{op_count}\(\)z{window_op}\({field}\)z{timeUnit}\({field}\)z{field}c3*K|] }|dz|f yw)z:{type}Nr0.0ps r4 z"parse_shorthand..~s)Oh1y=!*.s?A 581886o66> JsAA c3~K|]4}|j!|jj6ywr/)match groupdict)raexp shorthands r4rcz"parse_shorthand..s: yy#/ IIi * * ,s:=rrrIrCrrDr\rTrRrsort:z"{}" z0is not one of the valid encoding data types: {}.z, zi For more details, see https://altair-viz.github.io/user_guide/encodings/index.html#encoding-data-types. z>If you are trying to use a column name that contains a colon, zPprefix it with a backslash; for example "column\:name" instead of "column:name".)rYr^extendlist itertoolschainrdictnextINV_TYPECODE_MAPgetr6rErWrCObjectUnknownrrDr infer_vegalite_type_for_narwhalstuplerfindr-rgsplitjoin TYPECODE_MAPrZ)rnrparse_aggregatesparse_window_opsparse_timeunits parse_typesr^patternsregexpsattrsunescaped_fieldrZrCcolumns` r4parse_shorthandrsL/  H+,-234*+,234123 OOZL! )Oh)OPQG )T"   (,,U6]E&MJf  g&&&f UvU2"f U|D1.00r:..F f $_-Fo&  +&g&7&7&9: >v?O?O?Q Rf @ Hf %-/ %f a 0f %f a 0f 5 5> ! 'N5>//4q8 9T A NN5>//4R8 9@GG ,--/0 {  { O  O b  b   Lr6c|j}tj|r<|jj x}j sd|j fS|tjk(s&|tjk(s|tjk(ry|jry|tjk(s|tjk(ryd|}t|)NrArBrCrDzUnexpected DtypeKind: )r rEis_ordered_categoricalrget_categoriesis_emptyto_listString CategoricalBoolean is_numericrHrFr-)rr rr<s r4r}r}s LLE !!&)%zz88::DDF*,,...  Ubnn48K     "++ "''!1&ug.or6cHtdd}tdd}dfd }|S)a_ Use the signature and doc of ``tp`` for the decorated callable ``cb``. - **Overload 1**: Decorating method - **Overload 2**: Decorating function Returns ------- **Adding the annotation breaks typing**: Overload[Callable[[WrapsMethod[T, R]], WrappedMethod[T, P, R]], Callable[[WrapsFunc[R]], WrappedFunc[P, R]]] cyr/r0cbs r4decoratezuse_signature..decoratesFIr6cyr/r0rs r4rzuse_signature..decorates.decorates!Z4ZZ 6  H(9"++a&HILF&!P6+<+>> original = {"x": {"b": 2, "c": 4}} >>> update = {"x": {"b": 5, "d": 6}, "y": 40} >>> update_nested(original, update) # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} >>> original # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} )rr1rrrzrr)rrr)rrorig_vals r4rr sr> H%LLNS c7 #||C,H(N3 -h < # HSM# Or6ctj}|rddlm}|}nd}||j |yt j |y)Nr) get_ipython)sysexc_infoIPython.core.getipythonr showtraceback tracebackprint_exception) in_ipythonrrips r4display_tracebackr:s@||~H7 ]  ~ "!!8,r6)rdatumvalue _ChannelCache_CHANNEL_CACHEcLeZdZUded<ded<ed dZd dZd dZd dZy )rdict[type[SchemaBase], str]channel_to_name/dict[str, dict[_ChannelType, type[SchemaBase]]]name_to_channelc t}tS#t$rD|j|}t|_t |j|_|aYtSwxYwr/)r NameError__new___init_channel_to_namer_invert_group_channelsr)clscacheds r4 from_cachez_ChannelCache.from_cacheVs\ $#F   $[[%F%:%`] | type[``] | type[``], str] r)channels) altair.vegalite.v5.schemarFieldChannelMixinValueChannelMixinDatumChannelMixin__dict__rZrr issubclassr_encoding_name)chmixinscs r4rrs9  ! !2#7#79M9M MF##% %A a :a#8Z:=V 1  %  sAB cdd}t|jtd}|Dcic]\}}|||c}}Scc}}w)z;Grouped inverted index for `_ChannelCache.channel_to_name`.ci}|D]>\}}|j}|jdr$|jdrd}nd}|||<@|S)z Returns a 1-2 item dict, per channel. Never includes `datum`, as it is never utilized in `wrap_in_channel`. DatumValuerr)r:endswith)ititemr_rOsub_keys r4_reducez'_invert_group_channels.._reducesU -/EB;;D}}W%w'!!DM r6r)rzIterator[tuple[type[Any], str]]r8r )rr1r )mrgrouperrchanss r4rrsD $aggiA/G.5 6g(!UAwu~ g 66 6sActj}|D]c}t|ttfrt t |dn|}|jt|}||vr|||<Td|d}t||j|S)aa Infer typed keyword arguments for args and kwargs. Parameters ---------- args : Sequence Sequence of function args kwargs : MutableMapping Dict of function kwargs Returns ------- kwargs : dict All args and kwargs in a single dict, with keys and types based on the channels mapping. Nz encoding z specified twice.) rrrrtr~rxiterrrr-r)argsrcacheargrrr<s r4rrs"  $ $ &E&0tUm&DT$s)T "#%%d2h/ 6 !"F8 h\):;CS/ !  % %f --r6)rr r8z=InferredVegaLiteType | tuple[InferredVegaLiteType, list[Any]])rrr8r)rt.MutableMapping[Any, Any]r8r)r r r r r8r7)r9 pd.DataFramer8r)rnw.DataFrame[TIntoDataFrame]r8r)rrr8znw.DataFrame[Any])NTFTT)rnzdict[str, Any] | strrzIntoDataFrame | Nonerr7rr7rr7rr7r8r)rz nw.Seriesr8z8InferredVegaLiteType | tuple[InferredVegaLiteType, list])rzCallable[P, Any]).)rrrt.Mapping[Any, Any]r)zLiteral[False]r8r)rrrrr)z Literal[True]r8r)F)rr rrr)r7r8r)T)rr7)rrr8r)rztuple[Any, ...]rr)cr __future__rrurrerrrcollections.abcrrrr)rroperatorr typingr r r r rrrrnarwhals.stable.v1stablev1rEnarwhals.stable.v1.dependenciesrrnarwhals.stable.v1.typingraltair.utils.schemapirrr version_inforrrtyping_extensionsrrtr(r:raltair.utils._dfi_typesr r9!altair.vegalite.v5.schema._typingr!InferredVegaLiteTyper"r$r%r&r'r)r*r+r-rr1ry AGGREGATESWINDOW_AGGREGATES TIMEUNITSrtrvrVALID_TYPECODESrgrrh frozensetrrrrrr rArPr\rr}rrr _ChannelTyperrrr)rvs00r4rs" ==QQQT3CCwAALLw--82AX)? CL cN CL +xQ'7aT J M8AqD>1vN 5Aq6 Xk!Q$/23!Q H  &2%7%7%9:%9TQAqD%9: : R  hytL'94@P;QRS  ! !#((?"; <'$++CHHZ,@A$$SXXj;L.L%MN"))#((9*=>89L - -B-`$BAH & ! B*"&!"  x#x xx x  x  xxv =.'T %(% % % % %  %!% % % % %** * * *Z - 01  1 1 h07"7476.];s* K