translate_series

narwhals/__init__.py

@@ -1,7 +1,7 @@
 from narwhals import containers
-from narwhals.translate import to_original_object
 from narwhals.translate import translate_frame
+from narwhals.translate import translate_series
 
 __version__ = "0.1.13"
 
-__all__ = ["translate_frame", "to_original_object", "containers"]
+__all__ = ["translate_frame", "translate_series", "containers"]

narwhals/pandas_like/translate.py

@@ -9,10 +9,11 @@
     from narwhals.pandas_like.dataframe import DataFrame
     from narwhals.pandas_like.dataframe import LazyFrame
     from narwhals.pandas_like.namespace import Namespace
+    from narwhals.pandas_like.series import Series
 
 
 @overload
-def translate(
+def translate_frame(
     df: Any,
     implementation: str,
     *,
@@ -23,7 +24,7 @@ def translate(
 
 
 @overload
-def translate(
+def translate_frame(
     df: Any,
     implementation: str,
     *,
@@ -34,7 +35,7 @@ def translate(
 
 
 @overload
-def translate(
+def translate_frame(
     df: Any,
     implementation: str,
     *,
@@ -45,7 +46,7 @@ def translate(
 
 
 @overload
-def translate(
+def translate_frame(
     df: Any,
     implementation: str,
     *,
@@ -55,7 +56,7 @@ def translate(
     ...
 
 
-def translate(
+def translate_frame(
     df: Any,
     implementation: str,
     *,
@@ -76,4 +77,16 @@ def translate(
             df,
             implementation=implementation,
         )
-    return df, Namespace(implementation=df._implementation)
+    return df, Namespace(implementation=implementation)
+
+
+def translate_series(
+    series: Any,
+    implementation: str,
+) -> tuple[Series, Namespace]:
+    from narwhals.pandas_like.namespace import Namespace
+    from narwhals.pandas_like.series import Series
+
+    return Series(series, implementation=implementation), Namespace(
+        implementation=implementation
+    )

narwhals/translate.py

@@ -14,6 +14,7 @@
     from narwhals.spec import DataFrame
     from narwhals.spec import LazyFrame
     from narwhals.spec import Namespace
+    from narwhals.spec import Series
 
 
 @overload
@@ -86,34 +87,62 @@ def translate_frame(
             return LazyFrame(df), Namespace()
 
     if (pd := get_pandas()) is not None and isinstance(df, pd.DataFrame):
-        from narwhals.pandas_like.translate import translate
+        from narwhals.pandas_like.translate import translate_frame
 
-        return translate(
+        return translate_frame(
             df,
             implementation="pandas",
             eager_only=eager_only,
             lazy_only=lazy_only,
         )
 
     if (cudf := get_cudf()) is not None and isinstance(df, cudf.DataFrame):
-        from narwhals.pandas_like.translate import translate
+        from narwhals.pandas_like.translate import translate_frame
 
-        return translate(
+        return translate_frame(
             df, implementation="cudf", eager_only=eager_only, lazy_only=lazy_only
         )
 
     if (mpd := get_modin()) is not None and isinstance(df, mpd.DataFrame):
-        from narwhals.pandas_like.translate import translate
+        from narwhals.pandas_like.translate import translate_frame
 
-        return translate(
+        return translate_frame(
             df, implementation="modin", eager_only=eager_only, lazy_only=lazy_only
         )
 
     msg = f"Could not translate DataFrame {type(df)}, please open a feature request."
     raise TypeError(msg)
 
 
-def to_original_object(df: DataFrame | LazyFrame) -> Any:
-    if (pl := get_polars()) is not None and isinstance(df, (pl.DataFrame, pl.LazyFrame)):
-        return df
-    return df._dataframe  # type: ignore[union-attr]
+def translate_series(
+    series: Any,
+) -> tuple[Series, Namespace]:
+    if hasattr(series, "__narwhals_series__"):
+        return series.__narwhals_series__()  # type: ignore[no-any-return]
+
+    if (pl := get_polars()) is not None and isinstance(series, pl.Series):
+        from narwhals.polars import Namespace
+        from narwhals.polars import Series
+
+        return Series(series), Namespace()
+
+    if (pd := get_pandas()) is not None and isinstance(series, pd.DataFrame):
+        from narwhals.pandas_like.translate import translate_series
+
+        return translate_series(
+            series,
+            implementation="pandas",
+        )
+
+    if (cudf := get_cudf()) is not None and isinstance(series, cudf.DataFrame):
+        from narwhals.pandas_like.translate import translate_series
+
+        return translate_series(series, implementation="cudf")
+
+    if (mpd := get_modin()) is not None and isinstance(series, mpd.DataFrame):
+        from narwhals.pandas_like.translate import translate_series
+
+        return translate_series(series, implementation="modin")
+
+    msg = f"Could not translate DataFrame {type(series)}, please open a feature request."
+    raise TypeError(msg)