gddlmZddlZddlmZddlmZddlmZddlmZddlm Z er ddl Z ddl Z ddlmZdd lmZdd lmZdd lmZGd d ej*Z ddZGddZGddZy)) annotationsN) TYPE_CHECKING)Any)NoReturn)import_dtypes_module) parse_version)SelfInterchangeSeries)DType)Versionc(eZdZdZdZdZdZdZdZdZ y) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICALT/var/www/openai/venv/lib/python3.12/site-packages/narwhals/_interchange/dataframe.pyrrs% C D E D FHKr!rct|}|dtjk(rm|ddk(r|jS|ddk(r|j S|ddk(r|j S|ddk(r|j Sd}t||dtjk(rm|ddk(r|jS|ddk(r|jS|ddk(r|jS|ddk(r|jSd}t||dtjk(r=|ddk(r|jS|ddk(r|jSd }t||dtj k(r|j#S|dtj$k(r|j'S|dtj(k(r|j+S|dtj,k(r|j/Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )rrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32rBooleanrStringrDatetimer Categorical)interchange_dtypeversiondtypesmsgs r"'map_interchange_dtype_to_narwhals_dtyper; s"' *Fy}}, Q 2 %<<> ! Q 2 %<<> ! Q 2 %<<> ! Q 1 $;;= )S!!y~~- Q 2 %==? " Q 2 %==? " Q 2 %==? " Q 1 $<<> !*S!!y. Q 2 %>># # Q 2 %>># #+S!!y~~-~~y///}}y111  y444!!## !2 3 4C  r!ceZdZddZddZy)WrapInterchangeFramec||_yN_interchange_frame)selfinterchange_frames r"__init__zWrapInterchangeFrame.__init__Os "3r!c|jSr?r@rBs r" __dataframe__z"WrapInterchangeFrame.__dataframe__Rs&&&r!N)rCInterchangeFramereturnNone)rIrH)rrrrDrGr r!r"r=r=Ns 4'r!r=cdeZdZd dZd dZd dZddZddZddZddZ ddZ dd Z y )rHc<|j|_||_yr?)rGrA_version)rBdfr8s r"rDzInterchangeFrame.__init__Ws"$"2"2"4 r!cN|jt|j|S)Nr8) __class__r=rA)rBr8s r"_change_versionz InterchangeFrame._change_version[s(~~ !8!8 97  r!c|Sr?r rFs r"__narwhals_dataframe__z'InterchangeFrame.__narwhals_dataframe__`s r!cd}t|)NzCannot access native namespace for metadata-only dataframes with unknown backend.If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.)NotImplementedError)rBr:s r"__native_namespace__z%InterchangeFrame.__native_namespace__cs Y "#&&r!chddlm}||jj||jS)Nrr rP)narwhals._interchange.seriesr rAget_column_by_namerM)rBitemr s r" __getitem__zInterchangeFrame.__getitem__ks,B  # # 6 6t =1.5.0 to be installed, found )pandasr __version__api interchangefrom_dataframerArV)rBpdr:s r" to_pandaszInterchangeFrame.to_pandasrsc  (M',B B66%%44T5L5LM M99;8HJ &c* *r!c2ddlm}||jS)Nr)rb)pyarrow.interchangerbrA)rBrbs r"to_arrowzInterchangeFrame.to_arrow~s6d5566r!c H|dk(r`|jjDcic]<}|t|jj|j|j >c}S|dk(r#t |jjSd|d}t|cc}w)Nschemacolumnsz Attribute aG is not supported for metadata-only dataframes. Hint: you probably called `nw.from_native` on an object which isn't fully supported by Narwhals, yet implements `__dataframe__`. If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.)rA column_namesr;rZdtyperMlistrV)rBattr column_namer:s r" __getattr__zInterchangeFrame.__getattr__s 8  $(#:#:#G#G#I  $JK D++>>{KQQMM$J  Y //<<>? ?B B "#&&!sABc|std|Ds d}t||jj|}t |ds d}t||j |j |jS)Nc3<K|]}t|tywr?) isinstancestr).0xs r" z*InterchangeFrame.select..s!De*Q"4esz`select`-ing not by name is not supported for interchange-only level. If you would like to see this kind of object better supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues._dfzExpected interchange object to implement `_df` property to allow for recovering original object. See https://github.com/data-apis/dataframe-api/issues/360.rP)allrVrAselect_columns_by_namehasattrrQrxrM)rBexprs named_exprsr:frames r"selectzInterchangeFrame.selects c!De!DDF  &c* *''>>uEue$M &e, ,~~eii~??r!N)rNrr8r rIrJ)rBr r8r rIr )rIr)rBr rIr)r[rtrIr )rBr rIz pd.DataFrame)rBr rIzpa.Table)rnrtrIr)rBr r|rr}rrIr ) rrrrDrRrTrWr\rdrgrprr r!r"rHrHVsX  '  +7 '(@@@@  @r!rH)r7ztuple[DtypeKind, int, Any, Any]r8r rIr ) __future__renumtypingrrrnarwhals.utilsrrr^rcpyarrowpatyping_extensionsr rYr narwhals.dtypesr r IntEnumrr;r=rHr r!r"rsu" /(&>%& +6+AH+ +\''V@V@r!