IZjdddlmZddlZddlmZmZddlmZerhddlmZddl 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 gZ1ddZ2ddZ3ddZ4ddZ5ddZ6ddZ7ddZ8ddZ9ddZ:ddZ;ddZddZ?ddZ@ddZAdd!ZBdd#ZCdd&ZDdd'ZEdd)ZFdd*ZGdd-ZHdd/ZIdd0ZJdd2ZKdd3ZLdd5ZMdd7ZNdd8ZOdd:ZPdd;ZQdd=ZRdd@ZSddBZTddCZUddEZVddFZWddHZXddIZYddKZZddLZ[ddNZ\ddOZ]ddQZ^ddRZ_ddTZ`ddVZaddXZbddYZcdd[Zddd\Zedd^Zfdd`ZgddbZhdddZiddeZjddgZkddhZlddjZmddkZnddoZoddqZpddsZqddtZrddvZsddyZtddzZudd{Zvdd|Zwdd}Zxdd~ZyddZzddZ{ddZ|ddZ}ddZ~ddZddZddZddZddZddZddZddZgdZdS)) annotationsN) TYPE_CHECKINGAny) issue_warning) TypeGuard DataFrame)TypeIs)SQLFrameDataFrame)r LazyFrameSeries) IntoDataFrameTIntoLazyFrameT IntoSeriesTPandasLikeDType_1DArray _1DArrayInt_2DArray_NDArray _NumpyScalar_ShapeT fireducksreturnrcBtjddS)z;Get Polars module (if already imported - else return None).polarsNsysmodulesgeta/Users/jameslopez/projects/MentorCore/.venv/lib/python3.11/site-packages/narwhals/dependencies.py get_polarsr$1 ;??8T * **r"cBtjddS)z;Get pandas module (if already imported - else return None).pandasNrr!r"r# get_pandasr(6r%r"cBtjddS)zAGet modin.pandas module (if already imported - else return None).z modin.pandasNrr!r"r# get_modinr*;s ;??>4 0 00r"cBtjddS)z9Get cudf module (if already imported - else return None).cudfNrr!r"r#get_cudfr-@ ;??64 ( ((r"cBtjddS)z9Get cupy module (if already imported - else return None).cupyNrr!r"r#get_cupyr1Er.r"cBtjddS)zGet dask_expr module (if already imported - else return None).N dask_expr)r=hasattrrErrr )dds r# get_dask_exprrHhs@ """/GB 4L4L/| ;??; - --r"cBtjddS)zz'_is_pandas_dataframe..srVV   T22 24? 1 r3:/ 0 0VVVVVVr")r(ror any IMPORT_HOOKS)rjpdrrs` @r#_is_pandas_dataframerwst<< R , MB 1M1M RUVVVVV(VVVSSr"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. )rerwrjs r#is_pandas_dataframerz r"""  # ##r"TypeIs[pd.Series[Any]]ctx}durt|jp tfdtDS)z@Check whether `ser` is a pandas Series without importing pandas.Nc3K|]B}tj|dxduotjjVCdSrn)rrr ror'r)rprqrrrfs r#rsz$_is_pandas_series..srTT   T22 24? / sCJ- . .TTTTTTr")r(rorrtru)rfrvrrs` @r#_is_pandas_seriesrst<< R , KC1K1K PSTTTTT(TTTQQr"c>t|t|S)zCheck whether `ser` is a pandas Series without importing pandas. Warning: This method cannot be called on Narwhals Series. )rirrfs r#is_pandas_seriesr S!!! S ! !!r"indexTypeIs[pd.Index[Any]]ctx}durt|jp tfdtDS)zACheck whether `index` is a pandas Index without importing pandas.Nc3K|]B}tj|dxduotjjVCdSrn)rrr ror'Index)rprqrrrs r#rsz"is_pandas_index..srUU   T22 24? 0 ucj. / /UUUUUUr")r(rorrtru)rrvrrs` @r#is_pandas_indexrst<< R , LE281L1L QTUUUUU(UUURRr"TypeIs[mpd.DataFrame]cPtx}duot||jS)z@Check whether `df` is a modin DataFrame without importing modin.N)r*ror )rjmpds r#_is_modin_dataframers';; Ct + M 2s}0M0MMr"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. )rerrys r#is_modin_dataframer r""" r " ""r"TypeIs[mpd.Series]cPtx}duot||jS)z>Check whether `ser` is a modin Series without importing modin.N)r*ror)rfrs r#_is_modin_seriesrs';; Ct + K 3 0K0KKr"c>t|t|S)zCheck whether `ser` is a modin Series without importing modin. Warning: This method cannot be called on Narwhals Series. )rirrs r#is_modin_seriesrs S!!! C  r"TypeIs[mpd.Index[Any]]cPtx}duot||jS)z?Check whether `index` is a modin Index without importing modin.N)r*ror)rrs r#is_modin_indexrs';; Ct + L 5#)0L0LLr"TypeIs[cudf.DataFrame]cPtx}duot||jS)z>Check whether `df` is a cudf DataFrame without importing cudf.N)r-ror )rjr,s r#_is_cudf_dataframers'JJ Dt + N 2t~0N0NNr"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. )rerrys r#is_cudf_dataframer r""" b ! !!r"TypeIs[cudf.Series[Any]]cPtx}duot||jS)zt|t|S)zCheck whether `ser` is a cudf Series without importing cudf. Warning: This method cannot be called on Narwhals Series. )rirrs r#is_cudf_seriesrs S!!! 3  r"TypeIs[cudf.Index]cPtx}duot||jS)z=Check whether `index` is a cudf Index without importing cudf.N)r-ror)rr,s r# is_cudf_indexr s0JJ Dt +   tz11r"objboolcftx}duot||jo |jdkS)NrX)r1rondarraysize)rr0s r#is_cupy_scalarrs: D(  sDL ) )  HMr"TypeIs[dd.DataFrame]cPtx}duot||jS)z>Check whether `df` is a Dask DataFrame without importing Dask.N)r=ror )rjrGs r#_is_dask_dataframers)$&& &Bt 3 T 2r|8T8TTr"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. )rerrys r#is_dask_dataframer!rr"TypeIs[duckdb.DuckDBPyRelation]cPtx}duot||jS)zACheck whether `df` is a DuckDB Relation without importing DuckDB.N)r@roDuckDBPyRelation)rjr?s r#_is_duckdb_relationr+s1 ll "F4 / J F #55r"c>t|t|S)zCheck whether `df` is a DuckDB Relation without importing DuckDB. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. )rerrys r#is_duckdb_relationr2rr"TypeIs[ibis.Table]cdtx}duot||jjjS)z:Check whether `df` is a Ibis Table without importing Ibis.N)rCroexprtypesTable)rjrBs r#_is_ibis_tabler<s,JJ Dt + U 2ty?T0U0UUr"c>t|t|S)zCheck whether `df` is a Ibis Table without importing Ibis. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. )rerrys r# is_ibis_tablerAs r""" "  r"TypeIs[pl.DataFrame]cPtx}duot||jS)zBCheck whether `df` is a Polars DataFrame without importing Polars.N)r$ror rjpls r#_is_polars_dataframerK',, Bt + L 2r|0L0LLr"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. )rerrys r#is_polars_dataframerPr{r"TypeIs[pl.LazyFrame]cPtx}duot||jS)zBCheck whether `df` is a Polars LazyFrame without importing Polars.N)r$ror 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. )rerrys r#is_polars_lazyframer_r{r"TypeIs[pl.Series]cPtx}duot||jS)z@Check whether `ser` is a Polars Series without importing Polars.N)r$ror)rfrs r#_is_polars_seriesris',, Bt + J 3 0J0JJr"c>t|t|S)zCheck whether `ser` is a Polars Series without importing Polars. Warning: This method cannot be called on Narwhals Series. )rirrs r#is_polars_seriesrnrr"TypeIs[pl.Schema]cttx}o$t|dot||jS)NSchema)rr$rFrorrrs r#is_polars_schemarxs; :<< R  YWR%:%:Yz#ry?Y?Yr"TypeIs[pl.DataType]cfttx}ot||jSrn)rr$roDataTypers r#is_polars_data_typers* jll" # # D 3 (D(DDr"TypeIs[pa.ChunkedArray[Any]]cPtx}duot||jS)zHCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow.N)r4ro ChunkedArray)rfpas r#_is_pyarrow_chunked_arrayrs'-- B , QC1Q1QQr"c>t|t|S)zCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow. Warning: This method cannot be called on Narwhals Series. )rirrs r#is_pyarrow_chunked_arrayrs S!!! $S ) ))r"TypeIs[pa.Table]cPtx}duot||jS)z@Check whether `df` is a PyArrow Table without importing PyArrow.N)r4ror)rjrs r#_is_pyarrow_tablers'-- B , IB1I1IIr"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. )rerrys r#is_pyarrow_tablers r""" R  r"TypeIs[pa.Scalar[Any]]cPtx}duot||jSrn)r4roScalarrrs r#is_pyarrow_scalarrs'-- B , KC1K1KKr"TypeIs[pa.Schema]cfttx}ot||jSrn)rr4rorrs r#is_pyarrow_schemars* kmm# $ $ CC)C)CCr"TypeIs[pa.DataType]cfttx}ot||jSrn)rr4rorrs r#is_pyarrow_data_typers* kmm# $ $ EC)E)EEr"TypeIs[pyspark_sql.DataFrame]cjttx}duot||jS)zDCheck whether `df` is a PySpark DataFrame without importing PySpark.N)rrMror )rj pyspark_sqls r#_is_pyspark_dataframers< ')) )$6 2 r;0 1 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. )rerrys r#is_pyspark_dataframers r"""  $ $$r"TypeIs[PySparkConnectDataFrame]ctt) ddlm}n#t$rYdSwxYwt ||SdS)zLCheck whether `df` is a PySpark Connect DataFrame without importing PySpark.NrrF)rOpyspark.sql.connect.dataframer ImportErrorrorjr s r#_is_pyspark_connect_dataframers_(  ? ? ? ? ? ? ?   55 "i((( 5s  %%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. )rerrys r#is_pyspark_connect_dataframers r""" ( , ,,r"TypeIs[SQLFrameDataFrame]cNtddlm}t||SdS)zFCheck whether `df` is a SQLFrame DataFrame without importing SQLFrame.Nr) BaseDataFrameF)rRsqlframe.base.dataframerro)rjrs r#_is_sqlframe_dataframers3~~!999999"m,,, 5r"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. )rerrys r#is_sqlframe_dataframers r""" !" % %%r"arrAny | _NDArray[_ShapeT]TypeIs[_NDArray[_ShapeT]]cPtx}duot||jS)z=Check whether `arr` is a NumPy Array without importing NumPy.N)r8rorrnps r#is_numpy_arrayr s'++ Bd * Jz#rz/J/JJr"TypeIs[_1DArray]c6t|o |jdkS)z@Check whether `arr` is a 1D NumPy Array without importing NumPy.rXr ndimrs r#is_numpy_array_1dr #   038q=0r"TypeIs[_1DArrayInt]ctx}o.t|o||j|jSrn)r8r issubdtypedtypeintegerr s r#is_numpy_array_1d_intrs={{  1 c " " 1 MM#)RZ 0 0r"ctx}o.t|o||j|jSrn)r8rrrbool_r s r#is_numpy_array_1d_boolrs={{  / c " " / MM#)RX . .r"TypeIs[_2DArray]c6t|o |jdkS)z@Check whether `arr` is a 2D NumPy Array without importing NumPy.rrs r#is_numpy_array_2dr!rr"scalarTypeGuard[_NumpyScalar]cPtx}duot||jS)zACheck whether `scalar` is a NumPy Scalar without importing NumPy.N)r8rogeneric)r"r s r#is_numpy_scalarr&s) ++ Bd * Mz&"*/M/MMr"c\t|pt|pt|S)zCheck whether `df` is a pandas-like DataFrame without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )rwrrrys r#_is_pandas_like_dataframer(s/  # # X':2'>'> XBTUWBXBXXr"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. )rer(rys r#is_pandas_like_dataframer*s r""" $R ( ((r"c\t|pt|pt|S)zCheck whether `ser` is a pandas-like Series without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )rrrrs r#_is_pandas_like_seriesr,"s- S ! ! R%5c%:%: Roc>R>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. )rir,rs r#is_pandas_like_seriesr.*s S!!! !# & &&r"c\t|pt|pt|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_indexr06s. O."7"7O=;O;Or"TypeIs[PandasLikeDType]cttx}o1t||jjjt jfSrn)rr(roapi extensionsExtensionDtyper8rrrvs r#is_pandas_like_dtyper7@sE jll" # #   bf. 0A B))r"(TypeIs[pd.api.extensions.ExtensionDtype]cttx}o.t||jjjot |dS)Nto_arrow)rr(ror3r4r5rFr6s r# is_cudf_dtyper;FsI :<< R   % sRV.= ? ? % C $ $r"Any | IntoSeriesTTypeIs[IntoSeriesT]cddlm}||dp,t|pt|pt |S)Nr_hasattr_static__narwhals_series__)r^r@rrr,rr@s r#_is_native_seriesrCPs]////// 233 ' S ! ! ' $S ) ) ' "# & & r"Any | IntoDataFrameTTypeIs[IntoDataFrameT]cddlm}||dp,t|pt|pt |S)Nrr?__narwhals_dataframe__)r^r@rrr(rBs r#_is_native_dataframerH[s]////// 566 *  $ $ * S ! ! * %S ) ) r"Any | IntoLazyFrameTTypeIs[IntoLazyFrameT]cddlm}||dpht|pYt|pJt |p;t |p,t |pt|pt|S)Nrr?__narwhals_lazyframe__) r^r@rrrrrrrrBs r#_is_native_lazyframerMfs////// 566 '  $ $ ' c " " ' s # # ' #    ' ! % %  ' ) - -  ' "# & & r" native_seriescLddlm}t||pt|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.seriesrrorC)rNrs r#is_into_seriesrQus4.'&&&&& mV , , P0A-0P0PPr"native_dataframecLddlm}t||pt|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 rorH)rRr s r#is_into_dataframerU>.-,,,,, & 2 2 6J77r"native_lazyframecLddlm}t||pt|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 )rTr rorM)rWr s r#is_into_lazyframerZrVr"DataFrame[IntoDataFrameT] | Any!TypeIs[DataFrame[IntoDataFrameT]]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)rTr rors r#r\r\&-,,,,, b) $ $$r"lfAny | LazyFrame[IntoLazyFrameT]!TypeIs[LazyFrame[IntoLazyFrameT]]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`. rrY)rTr ro)r_r s r#r]r]r^r"Any | Series[IntoSeriesT]TypeIs[Series[IntoSeriesT]]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 )rPrro)rfrs r#rhrhs&'&&&&& c6 " ""r"cRt|o|jSrn)rhr is_integerrs r#is_narwhals_series_intrhs# c " " =sy';';'='==r"cRt|o|jSrn)rhr is_booleanrs r#is_narwhals_series_boolrks% c " " =sy';';'='==r"),r-r;r=r@rCr*r8r(r$r4rKrOrMrRrrrrrrrUrZrQrrrr\r]rhr rzrr*r0r.rrrrrrrrr)rr)rSrrrT)rfrrrT)rjrrrk)rfrrr|)rrrr)rjrrr)rfrrr)rrrr)rjrrr)rfrrr)rrrr)rrrr)rjrrr)rjrrr)rjrrr)rjrrr)rjrrr)rfrrr)rrrr)rrrr)rfrrr)rjrrr)rrrr)rrrr)rrrr)rjrrr)rjrrr)rjrrr)rrrr )rrrr)rrrr)rrrr)r"rrr#)rjrrr)rfrrr)rrrr)rrrr1)rrrr8)rr<rr=)rrDrrE)rrIrrJ)rNr<rr=)rRrDrrE)rWrIrrJ)rjr[rr\)r_r`rra)rfrcrrd) __future__rrtypingrrnarwhals._exceptionsrrr,dask.dataframe dataframerGr?rB modin.pandasr'rrvrrr3r pyspark.sqlsqlrrr PySparkConnectDataFrametyping_extensionsr narwhals._spark_like.dataframer rTr rPrnarwhals.typingrrrrrrrrrr frozensetrur$r(r*r-r1r4r8r;r=r@rCrHrKrMrOrRrerirwrzrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrr!r&r(r*r,r.r0r7r;rCrHrMrQrUrZr\r]rhrhrk__all__r!r"r#rzs#""""" %%%%%%%%......      KKKMMMKKK%%%%%%RRRRRR((((((@@@@@@77777777&&&&&&                        "y+'' ++++ ++++ 1111 )))) )))) ,,,, **** )))) 3333 ++++ )))) ....,,,, 0000 8888 ---- ( ( ( ( ( ( ( ($$$$""""NNNN ####LLLL !!!!MMMM OOOO """"MMMM     UUUU """"####VVVV MMMM $$$$MMMM $$$$KKKK """"EEEERRRR ****JJJJ !!!!LLLLDDDDFFFF%%%%----&&&&KKKK 1111 1111 NNNNYYYY ) ) ) )SSSS ' ' ' '     QQQQ8<< % % % % % % % % # # # #>>>>>>>> - - - r"