hhsLddlmZddlZddlmZmZddlmZerdddlZddl m Z ddl Z ddl Z ddlmZddlZddlZddlZddlmZddlmZddlmZmZddlm Z ddl!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/ZIdd1ZJdd3ZKdd5ZLdd8ZMdd:ZNdd<ZOdd>ZPdd@ZQddBZRddDZSddFZTddHZUddJZVddLZWddNZXddPZYddRZZddTZ[ddVZ\ddXZ]dd\Z^dd^Z_dd`Z`ddbZaddeZbddfZcddgZdddhZeddjZfddlZgddpZhddtZiddwZjdd{Zkdd~ZlddZmgdZndS)) annotationsN) TYPE_CHECKINGAny) issue_warning DataFrame) TypeGuardTypeIs)SQLFrameDataFrame)r LazyFrameSeries) IntoDataFrameTIntoLazyFrameT IntoSeriesTPandasLikeDType_1DArray _1DArrayInt_2DArray_NDArray _NumpyScalar_ShapeT fireducksreturnrcBtjddS)z;Get Polars module (if already imported - else return None).polarsNsysmodulesgetL/var/www/histauto/venv/lib/python3.11/site-packages/narwhals/dependencies.py get_polarsr$/ ;??8T * **r"cBtjddS)z;Get pandas module (if already imported - else return None).pandasNrr!r"r# get_pandasr(4r%r"cBtjddS)zAGet modin.pandas module (if already imported - else return None).z modin.pandasNrr!r"r# get_modinr*9s ;??>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_cupyr1Cr.r"cBtjddS)zGet dask_expr module (if already imported - else return None).N dask_expr)r=hasattrrErrr )dds r# get_dask_exprrHfs@ """/GB 4L4L/| ;??; - --r"cBtjddS)zz&is_pandas_dataframe..srVV   T22 24? 1 r3:/ 0 0VVVVVVr")r\r(rerany IMPORT_HOOKS)rSpdrhs` @r#is_pandas_dataframerms r"""<< R , MB 1M1M RUVVVVV(VVVSSr"TypeIs[pd.Series[Any]]cttx}durt|jp t fdt DS)zCheck whether `ser` is a pandas Series without importing pandas. Warning: This method cannot be called on Narwhals Series. Nc3K|]B}tj|dxduotjjVCdSrd)rrr rer'r)rfrgrhr]s r#riz#is_pandas_series..srTT   T22 24? / sCJ- . .TTTTTTr")r`r(rerrjrk)r]rlrhs` @r#is_pandas_seriesrqs S!!!<< R , KC1K1K PSTTTTT(TTTQQr"indexTypeIs[pd.Index[Any]]ctx}durt|jp tfdtDS)zACheck whether `index` is a pandas Index without importing pandas.Nc3K|]B}tj|dxduotjjVCdSrd)rrr rer'Index)rfrgrrrhs r#riz"is_pandas_index..srUU   T22 24? 0 ucj. / /UUUUUUr")r(rervrjrk)rrrlrhs` @r#is_pandas_indexrwst<< R , LE281L1L QTUUUUU(UUURRr"TypeIs[mpd.DataFrame]cnt|tx}duot||jS)zCheck whether `df` is a modin DataFrame without importing modin. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r\r*rer)rSmpds r#is_modin_dataframer{s7 r""";; Ct + M 2s}0M0MMr"TypeIs[mpd.Series]cnt|tx}duot||jS)zCheck whether `ser` is a modin Series without importing modin. Warning: This method cannot be called on Narwhals Series. N)r`r*rer)r]rzs r#is_modin_seriesr~s7 S!!!;; Ct + K 3 0K0KKr"TypeIs[mpd.Index[Any]]cPtx}duot||jS)z?Check whether `index` is a modin Index without importing modin.N)r*rerv)rrrzs r#is_modin_indexrs';; Ct + L 5#)0L0LLr"TypeIs[cudf.DataFrame]cnt|tx}duot||jS)zCheck whether `df` is a cudf DataFrame without importing cudf. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r\r-rer)rSr,s r#is_cudf_dataframers7 r"""JJ Dt + N 2t~0N0NNr"TypeIs[cudf.Series[Any]]cnt|tx}duot||jS)zCheck whether `ser` is a cudf Series without importing cudf. Warning: This method cannot be called on Narwhals Series. N)r`r-rer)r]r,s r#is_cudf_seriesrs7 S!!!JJ Dt + L 3 0L0LLr"TypeIs[cudf.Index]cPtx}duot||jS)z=Check whether `index` is a cudf Index without importing cudf.N)r-rerv)rrr,s r# is_cudf_indexrs0JJ Dt +   tz11r"objboolcftx}duot||jo |jdkS)N)r1rendarraysize)rr0s r#is_cupy_scalarrs: D(  sDL ) )  HMr"TypeIs[dd.DataFrame]cnt|tx}duot||jS)zCheck whether `df` is a Dask DataFrame without importing Dask. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r\r=rer)rSrGs r#is_dask_dataframers9 r"""$&& &Bt 3 T 2r|8T8TTr"TypeIs[duckdb.DuckDBPyRelation]cnt|tx}duot||jS)zCheck whether `df` is a DuckDB Relation without importing DuckDB. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r\r@reDuckDBPyRelation)rSr?s r#is_duckdb_relationrsA r""" ll "F4 / J F #55r"TypeIs[ibis.Table]ct|tx}duot||jjjS)zCheck whether `df` is a Ibis Table without importing Ibis. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r\rCreexprtypesTable)rSrBs r# is_ibis_tabler s< r"""JJ Dt + U 2ty?T0U0UUr"TypeIs[pl.DataFrame]cnt|tx}duot||jS)zCheck whether `df` is a Polars DataFrame without importing Polars. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r\r$rerrSpls r#is_polars_dataframer7 r""",, Bt + L 2r|0L0LLr"TypeIs[pl.LazyFrame]cnt|tx}duot||jS)zCheck whether `df` is a Polars LazyFrame without importing Polars. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r\r$rer rs r#is_polars_lazyframer rr"TypeIs[pl.Series]cnt|tx}duot||jS)zCheck whether `ser` is a Polars Series without importing Polars. Warning: This method cannot be called on Narwhals Series. N)r`r$rer)r]rs r#is_polars_seriesr*s7 S!!!,, Bt + J 3 0J0JJr"TypeIs[pl.Schema]cttx}o$t|dot||jS)NSchema)rr$rFrerrrs r#is_polars_schemar4s; :<< R  YWR%:%:Yz#ry?Y?Yr"TypeIs[pl.DataType]cfttx}ot||jSrd)rr$reDataTypers r#is_polars_data_typer;s* jll" # # D 3 (D(DDr"TypeIs[pa.ChunkedArray[Any]]cnt|tx}duot||jS)zCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow. Warning: This method cannot be called on Narwhals Series. N)r`r4re ChunkedArray)r]pas r#is_pyarrow_chunked_arrayr?s7 S!!!-- B , QC1Q1QQr"TypeIs[pa.Table]cnt|tx}duot||jS)zCheck whether `df` is a PyArrow Table without importing PyArrow. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r\r4rer)rSrs r#is_pyarrow_tablerIs7 r"""-- B , IB1I1IIr"TypeIs[pa.Scalar[Any]]cPtx}duot||jSrd)r4reScalarrrs r#is_pyarrow_scalarrSs'-- B , KC1K1KKr"TypeIs[pa.Schema]cfttx}ot||jSrd)rr4rerrs r#is_pyarrow_schemarWs* kmm# $ $ CC)C)CCr"TypeIs[pa.DataType]cfttx}ot||jSrd)rr4rerrs r#is_pyarrow_data_typer[s* kmm# $ $ EC)E)EEr"TypeIs[pyspark_sql.DataFrame]ct|ttx}duot||jS)zCheck whether `df` is a PySpark DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r\rrMrer)rS pyspark_sqls r#is_pyspark_dataframer_sL r""" ')) )$6 2 r;0 1 1  r"TypeIs[PySparkConnectDataFrame]ct|t) ddlm}n#t$rYdSwxYwt ||SdS)zCheck whether `df` is a PySpark Connect DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. NrrF)r\rOpyspark.sql.connect.dataframer ImportErrorrerSrs r#is_pyspark_connect_dataframerlso r"""(  ? ? ? ? ? ? ?   55 "i((( 5s & 44TypeIs[SQLFrameDataFrame]clt|tddlm}t ||SdS)zCheck whether `df` is a SQLFrame DataFrame without importing SQLFrame. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. Nr) BaseDataFrameF)r\rRsqlframe.base.dataframerre)rSrs r#is_sqlframe_dataframer|sC r"""~~!999999"m,,, 5r"arrAny | _NDArray[_ShapeT]TypeIs[_NDArray[_ShapeT]]cPtx}duot||jS)z=Check whether `arr` is a NumPy Array without importing NumPy.N)r8rerrnps r#is_numpy_arrayrs'++ Bd * Jz#rz/J/JJr"TypeIs[_1DArray]c6t|o |jdkS)z@Check whether `arr` is a 1D NumPy Array without importing NumPy.rrndimrs r#is_numpy_array_1dr #   038q=0r"TypeIs[_1DArrayInt]ctx}o.t|o||j|jSrd)r8r issubdtypedtypeintegerrs r#is_numpy_array_1d_intrs={{  1 c " " 1 MM#)RZ 0 0r"TypeIs[_2DArray]c6t|o |jdkS)z@Check whether `arr` is a 2D NumPy Array without importing NumPy.rrs r#is_numpy_array_2drrr"scalarTypeGuard[_NumpyScalar]cPtx}duot||jS)zACheck whether `scalar` is a NumPy Scalar without importing NumPy.N)r8regeneric)rrs r#is_numpy_scalarrs) ++ Bd * Mz&"*/M/MMr"czt|t|pt|pt|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. )r\rmr{r)rSs r#is_pandas_like_dataframers=r""" r " " U&8&<&< U@QRT@U@UUr"czt|t|pt|pt|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. )r`rqr~rr]s r#is_pandas_like_seriesrs:S!!! C OOC$8$8 ON3>> 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_series__)narwhals.seriesrrerFrrr)rrs r#is_into_seriesr sq.'&&&&& =&)) 0 ="7 8 8 0 M * * 0 $M 2 2 0 ! / / r"native_dataframeAny | IntoDataFrameTTypeIs[IntoDataFrameT]cddlm}t||p>> 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_dataframe__)narwhals.dataframerrerFrrr)r rs r#is_into_dataframersv.-,,,,, #Y// 6 #%= > > 6 / 0 0 6 , - - 6 $$4 5 5 r"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)rrrers r#rWrW&-,,,,, 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`. r)r )rr re)rr s r#rXrX-rr"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 )r rre)r]rs r#r_r_;s&'&&&&& c6 " ""r"cRt|o|jSrd)r_r is_integerrs r#is_narwhals_series_intr Gs# c " " =sy';';'='==r")r-rCr*r8r(r$r4rrrrrr r{r~rWrXr_rrmrwrrrqrrrrr)rr)rSrrrT)r]rrrT)rSrrra)r]rrrn)rrrrrs)rSrrrx)r]rrr|)rrrrr)rSrrr)r]rrr)rrrrr)rrrr)rSrrr)rSrrr)rSrrr)rSrrr)rSrrr)r]rrr)rrrr)rrrr)r]rrr)rSrrr)rrrr)rrrr)rrrr)rSrrr)rSrrr)rSrrr)rrrr)rrrr)rrrr)rrrr)rrrr)rSrrr)r]rrr)rrrrr)rrrr)rrrr)rrrr)r r rr)rSrrr)rrrr)r]rrr)o __future__rrtypingrrnarwhals._exceptionsrr,dask.dataframe dataframerGr?rB modin.pandasr'rzrlrrr3r pyspark.sqlsqlrrrPySparkConnectDataFrametyping_extensionsr r narwhals._spark_like.dataframer rr r rnarwhals.typingrrrrrrrrrr frozensetrkr$r(r*r-r1r4r8r;r=r@rCrHrKrMrOrRr\r`rmrqrwr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrWrXr_r __all__r!r"r#r/s#""""" %%%%%%%%......KKKMMMKKK%%%%%%RRRRRR33333333@@@@@@77777777&&&&&&                        "y+'' ++++ ++++ 1111 )))) )))) ,,,, **** )))) 3333 ++++ )))) ....,,,, 0000 8888 ---- ((((((((        NNNNLLLLMMMM OOOOMMMMUUUU    VVVVMMMMMMMMKKKKEEEERRRRJJJJLLLLDDDDFFFF            KKKK 1111 1111 NNNN V V V V P P P P DD % % % % % % % % # # # #>>>>   r"