diff --git a/narwhals/_spark_like/dataframe.py b/narwhals/_spark_like/dataframe.py index 31f16af55..3976bcfdd 100644 --- a/narwhals/_spark_like/dataframe.py +++ b/narwhals/_spark_like/dataframe.py @@ -85,15 +85,17 @@ def collect(self: Self) -> PandasLikeDataFrame: version=self._version, ) + def simple_select( + self: Self, + *exprs: str, + ) -> Self: + return self._from_native_frame(self._native_frame.select(*exprs)) + def select( self: Self, *exprs: SparkLikeExpr, **named_exprs: SparkLikeExpr, ) -> Self: - if exprs and all(isinstance(x, str) for x in exprs) and not named_exprs: - # This is a simple select - return self._from_native_frame(self._native_frame.select(*exprs)) - new_columns = parse_exprs_and_named_exprs(self, *exprs, **named_exprs) if not new_columns: