U )3g†'ã@sddlmZddlZddlmZddlmZer°ddlZejdkrPddlm Z n ddl m Z ddl Z ddl m ZddlZddlZddlmZddlZddlZddlZddlmZdd œd d „Zdd œd d „Zdd œdd„Zdd œdd„Zdd œdd„Zdd œdd„Z dd œdd„Z!dd œdd„Z"dd œdd„Z#dd œdd„Z$dd œdd„Z%dd œd d!„Z&dd"d#œd$d%„Z'dd&d'œd(d)„Z(dd*d+œd,d-„Z)dd.d#œd/d0„Z*dd1d'œd2d3„Z+dd4d+œd5d6„Z,dd7d#œd8d9„Z-dd:d'œd;d<„Z.dd=d+œd>d?„Z/dd@d#œdAdB„Z0ddCd#œdDdE„Z1ddFd#œdGdH„Z2ddId#œdJdK„Z3ddLd#œdMdN„Z4ddOd'œdPdQ„Z5ddRd'œdSdT„Z6ddUd#œdVdW„Z7ddXdYœdZd[„Z8dd\d#œd]d^„Z9dd\d'œd_d`„Z:dd\d+œdadb„Z;dcd\ddœdedf„ZdS)jé)Ú annotationsN)Ú TYPE_CHECKING)ÚAny)éé )Ú TypeGuard)Ú IntoSeriesr)ÚreturncCstj dd¡S)z;Get Polars module (if already imported - else return None).ÚpolarsN©ÚsysÚmodulesÚget©rrú9/tmp/pip-unpacked-wheel-hfsjijke/narwhals/dependencies.pyÚ get_polarssrcCstj dd¡S)z;Get pandas module (if already imported - else return None).ÚpandasNr rrrrÚ get_pandas"srcCs tj dd¡}dk r|jSdS)zAGet modin.pandas module (if already imported - else return None).ÚmodinN)r r rr)rrrrÚ get_modin'srcCstj dd¡S)z9Get cudf module (if already imported - else return None).ÚcudfNr rrrrÚget_cudf.srcCstj dd¡S)z9Get cupy module (if already imported - else return None).ZcupyNr rrrrÚget_cupy3srcCstj dd¡S)zGet dask_expr module (if already imported - else return None).Z dask_exprNr rrrrÚ get_dask_exprQsr!cCstj dd¡S)z9Get ibis module (if already imported - else return None).ÚibisNr rrrrÚget_ibisVsr#zTypeGuard[pd.DataFrame])Údfr cCstƒ}dk ot||jƒS)zBCheck whether `df` is a pandas DataFrame without importing pandas.N)rÚ isinstanceÚ DataFrame)r$ÚpdrrrÚis_pandas_dataframe[sr(zTypeGuard[pd.Series[Any]])Úserr cCstƒ}dk ot||jƒS)z@Check whether `ser` is a pandas Series without importing pandas.N)rr%ÚSeries)r)r'rrrÚis_pandas_series`sr+zTypeGuard[pd.Index])Úindexr cCstƒ}dk ot||jƒS)zACheck whether `index` is a pandas Index without importing pandas.N)rr%ÚIndex)r,r'rrrÚis_pandas_indexesr.zTypeGuard[mpd.DataFrame]cCstƒ}dk ot||jƒS)z@Check whether `df` is a modin DataFrame without importing modin.N)rr%r&)r$ÚmpdrrrÚis_modin_dataframejsr0zTypeGuard[mpd.Series]cCstƒ}dk ot||jƒS)z>Check whether `ser` is a modin Series without importing modin.N)rr%r*)r)r/rrrÚis_modin_seriesosr1zTypeGuard[mpd.Index]cCstƒ}dk ot||jƒS)z?Check whether `index` is a modin Index without importing modin.N)rr%r-)r,r/rrrÚis_modin_indextsÿr2zTypeGuard[cudf.DataFrame]cCstƒ}dk ot||jƒS)z>Check whether `df` is a cudf DataFrame without importing cudf.N)rr%r&)r$rrrrÚis_cudf_dataframe{sr3zTypeGuard[cudf.Series[Any]]cCstƒ}dk ot||jƒS)zCheck whether `df` is a Dask DataFrame without importing Dask.N)rr%r&)r$ÚddrrrÚis_dask_dataframeŒsr7z"TypeGuard[duckdb.DuckDBPyRelation]cCstƒ}dk ot||jƒS)zACheck whether `df` is a DuckDB Relation without importing DuckDB.N)r r%ZDuckDBPyRelation)r$rrrrÚis_duckdb_relation‘sÿr8zTypeGuard[ibis.Table]cCstƒ}dk ot||jjjƒS)z:Check whether `df` is a Ibis Table without importing Ibis.N)r#r%ÚexprÚtypesÚTable)r$r"rrrÚ is_ibis_table˜sr<zTypeGuard[pl.DataFrame]cCstƒ}dk ot||jƒS)zBCheck whether `df` is a Polars DataFrame without importing Polars.N)rr%r&©r$ÚplrrrÚis_polars_dataframesr?zTypeGuard[pl.LazyFrame]cCstƒ}dk ot||jƒS)zBCheck whether `df` is a Polars LazyFrame without importing Polars.N)rr%Z LazyFramer=rrrÚis_polars_lazyframe¢sr@zTypeGuard[pl.Series]cCstƒ}dk ot||jƒS)z@Check whether `ser` is a Polars Series without importing Polars.N)rr%r*)r)r>rrrÚis_polars_series§srAzTypeGuard[pa.ChunkedArray]cCstƒ}dk ot||jƒS)zHCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow.N)rr%Z ChunkedArray)r)ÚparrrÚis_pyarrow_chunked_array¬srCzTypeGuard[pa.Table]cCstƒ}dk ot||jƒS)z@Check whether `df` is a PyArrow Table without importing PyArrow.N)rr%r;)r$rBrrrÚis_pyarrow_table±srDzTypeGuard[np.ndarray])Úarrr cCstƒ}dk ot||jƒS)z=Check whether `arr` is a NumPy Array without importing NumPy.N)rr%Zndarray)rEÚnprrrÚis_numpy_array¶srGÚboolcCst|ƒpt|ƒpt|ƒS)z† Check whether `df` is a pandas-like DataFrame without doing any imports By "pandas-like", we mean: pandas, Modin, cuDF. )r(r0r3)r$rrrÚis_pandas_like_dataframe»srIcCst|ƒpt|ƒpt|ƒS)z„ Check whether `ser` is a pandas-like Series without doing any imports By "pandas-like", we mean: pandas, Modin, cuDF. )r+r1r4)r)rrrÚis_pandas_like_seriesÄsrJcCst|ƒpt|ƒpt|ƒS)z… Check whether `index` is a pandas-like Index without doing any imports By "pandas-like", we mean: pandas, Modin, cuDF. )r.r2r5)r,rrrÚis_pandas_like_indexÍsÿrKr)Ú native_seriesr cCs8ddlm}t||ƒp6t|dƒp6t|ƒp6t|ƒp6t|ƒS)a¼ Check 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 r)r*Z__narwhals_series__)Znarwhals.seriesr*r%ÚhasattrrArCrJ)rLr*rrrÚis_into_seriesØs  ÿþýûrN)Únative_dataframer cCs8ddlm}t||ƒp6t|dƒp6t|ƒp6t|ƒp6t|ƒS)a  Check 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 r)r&Z__narwhals_dataframe__)Znarwhals.dataframer&r%rMr?rDrI)rOr&rrrÚis_into_dataframeþs  ÿþýûrP)?Ú __future__rr ÚtypingrrrrFÚ version_inforZtyping_extensionsrZdask.dataframeZ dataframer6rr"Z modin.pandasrr/r'r r>rrBZnarwhals.typingrrrrrrrrrrr r!r#r(r+r.r0r1r2r3r4r5r7r8r<r?r@rArCrDrGrIrJrKrNrPÚ__all__rrrrÚsœ           &'ç