o LhNI@s*ddlmZddlZddlmZmZddlmZer|ddlZddl 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.e/d gZ0dddZ1dddZ2dddZ3dddZ4dddZ5dddZ6dddZ7dddZ8dddZ9dd d!Z:dd"d#Z;dd$d%Zdd*d+Z?dd,d-Z@dd0d1ZAdd3d4ZBdd6d7ZCdd9d:ZDdd=d>ZEdd@dAZFddCdDZGddFdGZHddIdJZIddLdMZJddOdPZKddSdTZLddVdWZMddYdZZNdd\d]ZOdd_d`ZPddbdcZQddedfZRddhdiZSddkdlZTddndoZUddqdrZVddtduZWddwdxZXdd|d}ZYdddZZdddZ[dddZ\dddZ]dddZ^dddZ_dddZ`dddZadddZbdddZcdddZddddZedddZfgdZgdS)) annotationsN) TYPE_CHECKINGAny) issue_warning DataFrame) TypeGuardTypeIs)SQLFrameDataFrame)r LazyFrameSeries) IntoDataFrameTIntoLazyFrameT IntoSeriesT_1DArray _1DArrayInt_2DArray_NDArray _NumpyScalar_ShapeT fireducksreturnrcCtjddS)z;Get Polars module (if already imported - else return None).polarsNsysmodulesgetrrV/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/narwhals/dependencies.py get_polars.r!cCr)z;Get pandas module (if already imported - else return None).pandasNrrrrr get_pandas3r"r$cCs tjdd}dur|jSdS)zAGet modin.pandas module (if already imported - else return None).modinN)rrrr#)r%rrr get_modin8sr&cCr)z9Get cudf module (if already imported - else return None).cudfNrrrrr get_cudf?r"r(cCr)z9Get cupy module (if already imported - else return None).cupyNrrrrr get_cupyDr"r*cCr)zGet dask_expr module (if already imported - else return None).N dask_expr)r1hasattrr6rrr)ddrrr get_dask_exprgsr9cCr)z   z&is_pandas_dataframe..)rHr$rOrany IMPORT_HOOKS)r@pdrrSr is_pandas_dataframe(rZTypeIs[pd.Series[Any]]cs:tt}durt|jptfddtDS)zCheck whether `ser` is a pandas Series without importing pandas. Warning: This method cannot be called on Narwhals Series. Nc3s4|]}tj|dduotjjVqdSrN)rrrrOr#r rPrTrIrr rUrVz#is_pandas_series..)rKr$rOr rWrX)rIrYrr]r is_pandas_seriesr[r^indexTypeIs[pd.Index[Any]]cs2t}dur t|jptfddtDS)zACheck whether `index` is a pandas Index without importing pandas.Nc3rMrN)rrrrOr#IndexrPr_rTrr rUrVz"is_pandas_index..)r$rOrarWrX)r_rYrrbr is_pandas_indexs(rcTypeIs[mpd.DataFrame]cC"t|t}duot||jS)zCheck whether `df` is a modin DataFrame without importing modin. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)rHr&rOr)r@mpdrrr is_modin_dataframergTypeIs[mpd.Series]cCre)zCheck whether `ser` is a modin Series without importing modin. Warning: This method cannot be called on Narwhals Series. N)rKr&rOr )rIrfrrr is_modin_seriesrhrjTypeIs[mpd.Index[Any]]cCt}duo t||jS)z?Check whether `index` is a modin Index without importing modin.N)r&rOra)r_rfrrr is_modin_indexrmTypeIs[cudf.DataFrame]cCre)zCheck whether `df` is a cudf DataFrame without importing cudf. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)rHr(rOr)r@r'rrr is_cudf_dataframerhrpTypeIs[cudf.Series[Any]]cCre)zCheck whether `ser` is a cudf Series without importing cudf. Warning: This method cannot be called on Narwhals Series. N)rKr(rOr )rIr'rrr is_cudf_seriesrhrrTypeIs[cudf.Index]cCrl)z=Check whether `index` is a cudf Index without importing cudf.N)r(rOra)r_r'rrr is_cudf_indexsrtobjboolcCs$t}duot||jo|jdkS)N)r*rOndarraysize)rur)rrr is_cupy_scalars  rzTypeIs[dd.DataFrame]cCre)zCheck whether `df` is a Dask DataFrame without importing Dask. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)rHr1rOr)r@r8rrr is_dask_dataframerhr|TypeIs[duckdb.DuckDBPyRelation]cCre)zCheck whether `df` is a DuckDB Relation without importing DuckDB. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)rHr3rODuckDBPyRelation)r@r2rrr is_duckdb_relationsrTypeIs[ibis.Table]cCs&t|t}duot||jjjS)zCheck whether `df` is a Ibis Table without importing Ibis. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)rHr5rOexprtypesTable)r@r4rrr is_ibis_table srTypeIs[pl.DataFrame]cCre)zCheck whether `df` is a Polars DataFrame without importing Polars. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)rHr!rOrr@plrrr is_polars_dataframerhrTypeIs[pl.LazyFrame]cCre)zCheck whether `df` is a Polars LazyFrame without importing Polars. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)rHr!rOr rrrr is_polars_lazyframe!rhrTypeIs[pl.Series]cCre)zCheck whether `ser` is a Polars Series without importing Polars. Warning: This method cannot be called on Narwhals Series. N)rKr!rOr )rIrrrr is_polars_series+rhrTypeIs[pa.ChunkedArray[Any]]cCre)zCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow. Warning: This method cannot be called on Narwhals Series. N)rKr,rO ChunkedArray)rIparrr is_pyarrow_chunked_array5rhrTypeIs[pa.Table]cCre)zCheck whether `df` is a PyArrow Table without importing PyArrow. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)rHr,rOr)r@rrrr is_pyarrow_table?rhrTypeIs[pa.Scalar[Any]]cCst}duo t||jSrN)r,rOScalar)rurrrr is_pyarrow_scalarIsrTypeIs[pyspark_sql.DataFrame]cCs&t|tt}duot||jS)zCheck whether `df` is a PySpark DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)rHrvr<rOr)r@ pyspark_sqlrrr is_pyspark_dataframeMs  rTypeIs[PySparkConnectDataFrame]cCsFt|tdur!zddlm}Wn tyYdSwt||SdS)zCheck whether `df` is a PySpark Connect DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. NrrF)rHr=pyspark.sql.connect.dataframer ImportErrorrOr@rrrr is_pyspark_connect_dataframeZs   rTypeIs[SQLFrameDataFrame]cCs,t|tdurddlm}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)rHr?sqlframe.base.dataframerrO)r@rrrr is_sqlframe_dataframejs    rarrAny | _NDArray[_ShapeT]TypeIs[_NDArray[_ShapeT]]cCrl)z=Check whether `arr` is a NumPy Array without importing NumPy.N)r.rOrxrnprrr is_numpy_arrayxrnrTypeIs[_1DArray]cCt|o|jdkS)z@Check whether `arr` is a 1D NumPy Array without importing NumPy.rwrndimrrrr is_numpy_array_1d}rTypeIs[_1DArrayInt]cCs"t}ot|o||j|jSrN)r.r issubdtypedtypeintegerrrrr is_numpy_array_1d_ints rTypeIs[_2DArray]cCr)z@Check whether `arr` is a 2D NumPy Array without importing NumPy.rrrrr is_numpy_array_2drrscalarTypeGuard[_NumpyScalar]cCrl)zACheck whether `scalar` is a NumPy Scalar without importing NumPy.N)r.rOgeneric)rrrrr is_numpy_scalarsrcC t|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. )rHrZrgrp)r@rrr is_pandas_like_dataframercCr)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. )rKr^rjrrrIrrr is_pandas_like_seriesrrcCst|p t|p t|S)zCheck whether `index` is a pandas-like Index without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. )rcrmrt)r_rrr is_pandas_like_indexsr native_seriesAny | IntoSeriesTTypeIs[IntoSeriesT]cC8ddlm}t||pt|dpt|pt|pt|S)aCheck whether `native_series` can be converted to a Narwhals Series. Arguments: native_series: The object to check. Returns: `True` if `native_series` can be converted to a Narwhals Series, `False` otherwise. 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_series__)narwhals.seriesr rOr7rrr)rr rrr is_into_series  rnative_dataframeAny | IntoDataFrameTTypeIs[IntoDataFrameT]cCr)aCheck whether `native_dataframe` can be converted to a Narwhals DataFrame. Arguments: native_dataframe: The object to check. Returns: `True` if `native_dataframe` can be converted to a Narwhals DataFrame, `False` otherwise. 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_dataframe__)narwhals.dataframerrOr7rrr)rrrrr is_into_dataframerrDataFrame[IntoDataFrameT] | Any!TypeIs[DataFrame[IntoDataFrameT]]cCddlm}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)rrrOrrrr rC rClfAny | LazyFrame[IntoLazyFrameT]!TypeIs[LazyFrame[IntoLazyFrameT]]cCr)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 rO)rr rrr rDrrDAny | Series[IntoSeriesT]TypeIs[Series[IntoSeriesT]]cCr)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 )rr rO)rIr rrr rJs  rJcCst|o|jSrN)rJr is_integerrrrr is_narwhals_series_int+sr)r(r5r&r.r$r!r,rprrr|rrrrgrjrCrDrJrrZrcrrr^rrrrr)rr)r@rrrA)rIrrrA)r@rrrL)rIrrr\)r_rrr`)r@rrrd)rIrrri)r_rrrk)r@rrro)rIrrrq)r_rrrs)rurrrv)r@rrr{)r@rrr})r@rrr)r@rrr)r@rrr)rIrrr)rIrrr)r@rrr)rurrr)r@rrr)r@rrr)r@rrr)rrrr)rrrr)rrrr)rrrr)rrrr)r@rrrv)rIrrrv)r_rrrv)rrrr)rrrr)r@rrr)rrrr)rIrrr)h __future__rrtypingrrnarwhals._exceptionsrr'dask.dataframe dataframer8r2r4 modin.pandasr#rfrYrrr+r pyspark.sqlsqlrrrPySparkConnectDataFrametyping_extensionsrr narwhals._spark_like.dataframer rr rr narwhals.typingrrrrrrrrr frozensetrXr!r$r&r(r*r,r.r0r1r3r5r9r;r<r=r?rHrKrZr^rcrgrjrmrprrrtrzr|rrrrrrrrrrrrrrrrrrrrrrCrDrJr__all__rrrr s        ,                                % %