diff --git a/spec/API_specification/dataframe_api/dataframe_object.py b/spec/API_specification/dataframe_api/dataframe_object.py index 475797ab..e7899260 100644 --- a/spec/API_specification/dataframe_api/dataframe_object.py +++ b/spec/API_specification/dataframe_api/dataframe_object.py @@ -109,13 +109,13 @@ def get_column_by_name(self, name: str, /) -> Column: """ ... - def select(self, names: Sequence[str], /) -> Self: + def select(self, *names: str) -> Self: """ Select multiple columns by name. Parameters ---------- - names : Sequence[str] + *names : str Returns ------- diff --git a/spec/API_specification/examples/tpch/q5.py b/spec/API_specification/examples/tpch/q5.py index cdca0806..b69b8300 100644 --- a/spec/API_specification/examples/tpch/q5.py +++ b/spec/API_specification/examples/tpch/q5.py @@ -68,7 +68,7 @@ def query( * (1 - result.get_column_by_name("l_discount")) ).rename("revenue") result = result.assign(new_column) - result = result.select(["revenue", "n_name"]) + result = result.select("revenue", "n_name") result = result.group_by("n_name").sum() return result.dataframe