O3jdddlmZddlZddlmZmZddlmZerhddlmZddl Z ddl m Z ddl Z ddlZddlmZddlZddlZddlZddlmZddlmZddlmZddlm Z dd l!mZm"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/e0d gZ1dcd Z2dcdZ3dcdZ4dcdZ5dcdZ6dcdZ7dcdZ8dcdZ9dcdZ:dcdZ;dcdZdcdZ?dcdZ@dcdZAdddZBdedZCdfdZDdfd ZEdgd!ZFdgd"ZGdhd#ZHdid$ZIdid%ZJdjd&ZKdjd'ZLdkd(ZMdld)ZNdld*ZOdmd+ZPdmd,ZQdnd-ZRdod.ZSdpd/ZTdpd0ZUdqd1ZVdqd2ZWdrd3ZXdrd4ZYdsd5ZZdsd6Z[dtd7Z\dtd8Z]dud9Z^dud:Z_dvd;Z`dwd<Zadxd=Zbdxd>Zcdyd?Zddyd@ZedzdAZfd{dBZgd|dCZhd}dDZid}dEZjd~dFZkd~dGZlddHZmddIZnddJZoddKZpddLZqddMZrddNZsddOZtddPZuddQZvddRZwddSZxddTZyddUZz ddVZ{ddWZ|ddXZ}ddYZ~ddZZdd[Zdd\Z dd]Z dd^Zdd_Zdd`Z ddaZgdbZy)) annotationsN) TYPE_CHECKINGAny) issue_warning) TypeGuard DataFrame)TypeIs)SQLFrameDataFrame)r LazyFrameSeries) IntoDataFrameTIntoLazyFrameT IntoSeriesTPandasLikeDType_1DArray _1DArrayInt_2DArray_NDArray _NumpyScalar_ShapeT fireduckscBtjjddS)z;Get Polars module (if already imported - else return None).polarsNsysmodulesgetB/DATA/.local/lib/python3.12/site-packages/narwhals/dependencies.py get_polarsr#1 ;;??8T **r!cBtjjddS)z;Get pandas module (if already imported - else return None).pandasNrr r!r" get_pandasr'6r$r!cBtjjddS)zAGet modin.pandas module (if already imported - else return None).z modin.pandasNrr r!r" get_modinr);s ;;??>4 00r!cBtjjddS)z9Get cudf module (if already imported - else return None).cudfNrr r!r"get_cudfr,@ ;;??64 ((r!cBtjjddS)z9Get cupy module (if already imported - else return None).cupyNrr r!r"get_cupyr0Er-r!cBtjjddS)zGet dask_expr module (if already imported - else return None).N dask_expr)r<hasattrrDrrr)dds r" get_dask_exprrGhs8 ""/GB 4L|| ;;??; --r!cBtjjddS)zz'_is_pandas_dataframe..sSV(K  T2 24? 1 r3::// 0 1'AA)r'rkr any IMPORT_HOOKS)rnpdros` @r"_is_pandas_dataframerusF< R , MB 1M RUV(VSr!c.t|t|S)zCheck whether `df` is a pandas DataFrame without importing pandas. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrurns r"is_pandas_dataframerx r"  ##r!ctx}duxrt|jxstfdtDS)z@Check whether `ser` is a pandas Series without importing pandas.Nc3K|]J}tjj|dxduxr tjj Lywrj)rrrrkr&r)rlrmrorfs r"rpz$_is_pandas_series..sST(K  T2 24? / sCJJ-- . /'rq)r'rkrrrrs)rfrtros` @r"_is_pandas_seriesr|sF< R , KC1K PST(TQr!c.t|t|S)zCheck whether `ser` is a pandas Series without importing pandas. Warning: This method cannot be called on Narwhals Series. )rgr|rfs r"is_pandas_seriesr S! S !!r!ctx}duxrt|jxstfdtDS)zACheck whether `index` is a pandas Index without importing pandas.Nc3K|]J}tjj|dxduxr tjj Lywrj)rrrrkr&Index)rlrmindexros r"rpz"is_pandas_index..sSU(K  T2 24? 0 ucjj.. / 0'rq)r'rkrrrrs)rrtros` @r"is_pandas_indexrsF< R , LE2881L QTU(URr!cNtx}duxrt||jS)z@Check whether `df` is a modin DataFrame without importing modin.N)r)rkr )rnmpds r"_is_modin_dataframers$; Ct + M 2s}}0MMr!c.t|t|S)zCheck whether `df` is a modin DataFrame without importing modin. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_modin_dataframer r" r ""r!cNtx}duxrt||jS)z>Check whether `ser` is a modin Series without importing modin.N)r)rkr)rfrs r"_is_modin_seriesrs$; Ct + K 3 0KKr!c.t|t|S)zCheck whether `ser` is a modin Series without importing modin. Warning: This method cannot be called on Narwhals Series. )rgrr~s r"is_modin_seriesrs S! C  r!cNtx}duxrt||jS)z?Check whether `index` is a modin Index without importing modin.N)r)rkr)rrs r"is_modin_indexrs$; Ct + L 5#))0LLr!cNtx}duxrt||jS)z>Check whether `df` is a cudf DataFrame without importing cudf.N)r,rkr )rnr+s r"_is_cudf_dataframers$J Dt + N 2t~~0NNr!c.t|t|S)zCheck whether `df` is a cudf DataFrame without importing cudf. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_cudf_dataframer r" b !!r!cNtx}duxrt||jS)zCheck whether `df` is a Dask DataFrame without importing Dask.N)r<rkr )rnrFs r"_is_dask_dataframers%$& &Bt 3 T 2r||8TTr!c.t|t|S)zCheck whether `df` is a Dask DataFrame without importing Dask. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_dask_dataframer!rr!cNtx}duxrt||jS)zACheck whether `df` is a DuckDB Relation without importing DuckDB.N)r?rkDuckDBPyRelation)rnr>s r"_is_duckdb_relationr+s- l "F4 / J F # #5r!c.t|t|S)zCheck whether `df` is a DuckDB Relation without importing DuckDB. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. )rcrrws r"is_duckdb_relationr2rr!cvtx}duxr*t||jjjS)z:Check whether `df` is a Ibis Table without importing Ibis.N)rBrkexprtypesTable)rnrAs r"_is_ibis_tabler<s.J Dt + U 2tyy?T?T0UUr!c.t|t|S)zCheck whether `df` is a Ibis Table without importing Ibis. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. )rcrrws r" is_ibis_tablerAs r" " r!cNtx}duxrt||jS)zBCheck whether `df` is a Polars DataFrame without importing Polars.N)r#rkr rnpls r"_is_polars_dataframerK$, Bt + L 2r||0LLr!c.t|t|S)zCheck whether `df` is a Polars DataFrame without importing Polars. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_polars_dataframerPryr!cNtx}duxrt||jS)zBCheck whether `df` is a Polars LazyFrame without importing Polars.N)r#rkr rs r"_is_polars_lazyframerZrr!c.t|t|S)zCheck whether `df` is a Polars LazyFrame without importing Polars. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. )rcrrws r"is_polars_lazyframer_ryr!cNtx}duxrt||jS)z@Check whether `ser` is a Polars Series without importing Polars.N)r#rkr)rfrs r"_is_polars_seriesris$, Bt + J 3 0JJr!c.t|t|S)zCheck whether `ser` is a Polars Series without importing Polars. Warning: This method cannot be called on Narwhals Series. )rgrr~s r"is_polars_seriesrnrr!cxttx}xr$t|dxrt||jS)NSchema)boolr#rErkrrrs r"is_polars_schemarxs2 :< R YWR%:Yz#ryy?Yr!c\ttx}xrt||jSrj)rr#rkDataTypers r"is_polars_data_typers$ jl" # D 3 (DDr!cNtx}duxrt||jS)zHCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow.N)r3rk ChunkedArray)rfpas r"_is_pyarrow_chunked_arrayrs$- B , QC1QQr!c.t|t|S)zCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow. Warning: This method cannot be called on Narwhals Series. )rgrr~s r"is_pyarrow_chunked_arrayrs S! $S ))r!cNtx}duxrt||jS)z@Check whether `df` is a PyArrow Table without importing PyArrow.N)r3rkr)rnrs r"_is_pyarrow_tablers$- B , IB1IIr!c.t|t|S)zCheck whether `df` is a PyArrow Table without importing PyArrow. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_pyarrow_tablers r" R  r!cNtx}duxrt||jSrj)r3rkScalarrrs r"is_pyarrow_scalarrs$- B , KC1KKr!c\ttx}xrt||jSrj)rr3rkrrs r"is_pyarrow_schemars$ km# $ CC)CCr!c\ttx}xrt||jSrj)rr3rkrrs r"is_pyarrow_data_typers$ km# $ EC)EEr!c`ttx}duxrt||jS)zDCheck whether `df` is a PySpark DataFrame without importing PySpark.N)rrLrkr )rn pyspark_sqls r"_is_pyspark_dataframers4 ') )$6 2 r;00 1 r!c.t|t|S)zCheck whether `df` is a PySpark DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_pyspark_dataframers r"  $$r!c\t ddlm}t ||Sy#t$rYywxYw)zLCheck whether `df` is a PySpark Connect DataFrame without importing PySpark.rrF)rNpyspark.sql.connect.dataframer ImportErrorrkrnr s r"_is_pyspark_connect_dataframers9(  ?"i((   s  ++c.t|t|S)zCheck whether `df` is a PySpark Connect DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_pyspark_connect_dataframers r" ( ,,r!c<tddlm}t||Sy)zFCheck whether `df` is a SQLFrame DataFrame without importing SQLFrame.r) BaseDataFrameF)rQsqlframe.base.dataframerrk)rnrs r"_is_sqlframe_dataframers~!9"m,, r!c.t|t|S)zCheck whether `df` is a SQLFrame DataFrame without importing SQLFrame. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_sqlframe_dataframers r" !" %%r!cNtx}duxrt||jS)z=Check whether `arr` is a NumPy Array without importing NumPy.N)r7rkrarrnps r"is_numpy_arrayrs$+ Bd * Jz#rzz/JJr!c:t|xr|jdk(S)z@Check whether `arr` is a 1D NumPy Array without importing NumPy.rUrndimrs r"is_numpy_array_1dr #  0388q=0r!ctx}xr3t|xr&|j|j|jSrj)r7r issubdtypedtypeintegerrs r"is_numpy_array_1d_intrs9{  1 c " 1 MM#))RZZ 0r!ctx}xr3t|xr&|j|j|jSrj)r7rrrbool_rs r"is_numpy_array_1d_boolrs9{  / c " / MM#))RXX .r!c:t|xr|jdk(S)z@Check whether `arr` is a 2D NumPy Array without importing NumPy.rrs r"is_numpy_array_2drrr!cNtx}duxrt||jS)zACheck whether `scalar` is a NumPy Scalar without importing NumPy.N)r7rkgeneric)scalarrs r"is_numpy_scalarrs& + Bd * Mz&"**/MMr!cLt|xst|xs t|S)zCheck whether `df` is a pandas-like DataFrame without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )rurrrws r"_is_pandas_like_dataframers&  # X':2'> XBTUWBXXr!c.t|t|S)zCheck whether `df` is a pandas-like DataFrame without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )rcrrws r"is_pandas_like_dataframer sr" $R ((r!cLt|xst|xs t|S)zCheck whether `ser` is a pandas-like Series without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )r|rrr~s r"_is_pandas_like_seriesr "s$ S ! R%5c%: Roc>RRr!c.t|t|S)zCheck whether `ser` is a pandas-like Series without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. Warning: This method cannot be called on Narwhals Series. )rgr r~s r"is_pandas_like_seriesr*sS! !# &&r!cLt|xst|xs t|S)zCheck whether `index` is a pandas-like Index without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )rrr)rs r"is_pandas_like_indexr6s% O."7O=;Or!cttx}xr>t||jjj t jfSrj)rr'rkapi extensionsExtensionDtyper7rrrts r"is_pandas_like_dtyper@sD jl" #   bff.. 0A0A B)r!cttx}xr8t||jjj xr t |dS)Nto_arrow)rr'rkrrrrErs r" is_cudf_dtyperFsF :< R  % sRVV..== ? % C $r!cnddlm}||dxs%t|xst|xs t |S)Nr_hasattr_static__narwhals_series__)r[rrrr rrs r"_is_native_seriesrPs@/ 23 ' S ! ' $S ) ' "# & r!cnddlm}||dxs%t|xst|xs t |S)Nrr__narwhals_dataframe__)r[rrrrrs r"_is_native_dataframer"[s@/ 56 *  $ * S ! * %S ) r!cddlm}||dxsYt|xsLt|xs?t |xs2t |xs%t |xst|xs t|S)Nrr__narwhals_lazyframe__) r[rrrrrrrrrs r"_is_native_lazyframer%fs|/ 56 '  $ ' c " ' s # ' #   ' ! %  ' ) -  ' "# & r!c@ddlm}t||xs t|S)aMCheck whether `native_series` can be converted to a Narwhals Series. Arguments: native_series: The object to check. Examples: >>> import pandas as pd >>> import polars as pl >>> import numpy as np >>> import narwhals as nw >>> s_pd = pd.Series([1, 2, 3]) >>> s_pl = pl.Series([1, 2, 3]) >>> np_arr = np.array([1, 2, 3]) >>> nw.dependencies.is_into_series(s_pd) True >>> nw.dependencies.is_into_series(s_pl) True >>> nw.dependencies.is_into_series(np_arr) False rr )narwhals.seriesrrkr) native_seriesrs r"is_into_seriesr)us.' mV , P0A-0PPr!c@ddlm}t||xs t|S)aCheck whether `native_dataframe` can be converted to a Narwhals DataFrame. Arguments: native_dataframe: The object to check. Examples: >>> import pandas as pd >>> import polars as pl >>> import numpy as np >>> from narwhals.dependencies import is_into_dataframe >>> df_pd = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> df_pl = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> np_arr = np.array([[1, 4], [2, 5], [3, 6]]) >>> is_into_dataframe(df_pd) True >>> is_into_dataframe(df_pl) True >>> is_into_dataframe(np_arr) False rr)narwhals.dataframer rkr")native_dataframer s r"is_into_dataframer-'.- & 2 6J7r!c@ddlm}t||xs t|S)aCheck whether `native_lazyframe` can be converted to a Narwhals LazyFrame. Arguments: native_lazyframe: The object to check. Examples: >>> import pandas as pd >>> import polars as pl >>> import numpy as np >>> from narwhals.dependencies import is_into_lazyframe >>> df_pd = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> lf_pl = pl.LazyFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> np_arr = np.array([[1, 4], [2, 5], [3, 6]]) >>> is_into_lazyframe(df_pd) False >>> is_into_lazyframe(lf_pl) True >>> is_into_lazyframe(np_arr) False rr )r+r rkr%)native_lazyframer s r"is_into_lazyframer2r.r!c&ddlm}t||S)zCheck whether `df` is a Narwhals DataFrame. This is useful if you expect a user to pass in a Narwhals DataFrame directly, and you want to catch both `narwhals.DataFrame` and `narwhals.stable.v1.DataFrame`. rr)r+r rkrs r"rYrY- b) $$r!c&ddlm}t||S)zCheck whether `lf` is a Narwhals LazyFrame. This is useful if you expect a user to pass in a Narwhals LazyFrame directly, and you want to catch both `narwhals.LazyFrame` and `narwhals.stable.v1.LazyFrame`. rr0)r+r rk)lfr s r"rZrZr4r!c&ddlm}t||S)zCheck whether `ser` is a Narwhals Series. This is useful if you expect a user to pass in a Narwhals Series directly, and you want to catch both `narwhals.Series` and `narwhals.stable.v1.Series`. rr )r'rrk)rfrs r"reres' c6 ""r!cPt|xr|jjSrj)rer is_integerr~s r"is_narwhals_series_intr:s c " =syy';';'==r!cPt|xr|jjSrj)rer is_booleanr~s r"is_narwhals_series_boolr=s! c " =syy';';'==r!),r,r:r<r?rBr)r7r'r#r3rJrNrLrQrrrrrrr-r2r)rrrrYrZrerrxrr rrrrrrrrrrr)returnr)r`rr>None)rfrr>r?)rnrr>zTypeIs[pd.DataFrame])rfrr>zTypeIs[pd.Series[Any]])rrr>zTypeIs[pd.Index[Any]])rnrr>zTypeIs[mpd.DataFrame])rfrr>zTypeIs[mpd.Series])rrr>zTypeIs[mpd.Index[Any]])rnrr>zTypeIs[cudf.DataFrame])rfrr>zTypeIs[cudf.Series[Any]])rrr>zTypeIs[cudf.Index])rrr>r)rnrr>zTypeIs[dd.DataFrame])rnrr>zTypeIs[duckdb.DuckDBPyRelation])rnrr>zTypeIs[ibis.Table])rnrr>zTypeIs[pl.DataFrame])rnrr>zTypeIs[pl.LazyFrame])rfrr>zTypeIs[pl.Series])rrr>zTypeIs[pl.Schema])rrr>zTypeIs[pl.DataType])rfrr>zTypeIs[pa.ChunkedArray[Any]])rnrr>zTypeIs[pa.Table])rrr>zTypeIs[pa.Scalar[Any]])rrr>zTypeIs[pa.Schema])rrr>zTypeIs[pa.DataType])rnrr>zTypeIs[pyspark_sql.DataFrame])rnrr>zTypeIs[PySparkConnectDataFrame])rnrr>zTypeIs[SQLFrameDataFrame])rzAny | _NDArray[_ShapeT]r>zTypeIs[_NDArray[_ShapeT]])rrr>zTypeIs[_1DArray])rrr>zTypeIs[_1DArrayInt])rrr>zTypeIs[_2DArray])rrr>zTypeGuard[_NumpyScalar])rnrr>r)rfrr>r)rrr>r)rrr>zTypeIs[PandasLikeDType])rrr>z(TypeIs[pd.api.extensions.ExtensionDtype])rAny | IntoSeriesTr>TypeIs[IntoSeriesT])rAny | IntoDataFrameTr>TypeIs[IntoDataFrameT])rAny | IntoLazyFrameTr>TypeIs[IntoLazyFrameT])r(r@r>rA)r,rBr>rC)r1rDr>rE)rnzDataFrame[IntoDataFrameT] | Anyr>z!TypeIs[DataFrame[IntoDataFrameT]])r6zAny | LazyFrame[IntoLazyFrameT]r>z!TypeIs[LazyFrame[IntoLazyFrameT]])rfzAny | Series[IntoSeriesT]r>zTypeIs[Series[IntoSeriesT]]) __future__rrtypingrrnarwhals._exceptionsrrr+dask.dataframe dataframerFr>rA modin.pandasr&rrtrrr2r pyspark.sqlsqlrrr PySparkConnectDataFrametyping_extensionsr narwhals._spark_like.dataframer r+r r'rnarwhals.typingrrrrrrrrrr frozensetrsr#r'r)r,r0r3r7r:r<r?rBrGrJrLrNrQrcrgrurxr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrr"r%r)r-r2rYrZrer:r=__all__r r!r"rTse# %. %R(@7&   "+' + + 1 ) ) , * ) 3 + ) ., 0 8 - ( ($"N #L !M O "M  U "#V M $M $K "ER *J !LDF%-&K 1 1 NY )S '  - Q8<< %' %& % %' %& % #>> "> > - r!