Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/bigframes/bigframes/bigquery/_operations/ml.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,12 @@
import google.cloud.bigquery
import pandas as pd

import bigframes.core.col as col
import bigframes.core.logging.log_adapter as log_adapter
import bigframes.core.sql.ml
import bigframes.dataframe as dataframe
import bigframes.ml.base
import bigframes.session
Comment thread
sycai marked this conversation as resolved.
import bigframes.core.col as col
from bigframes.bigquery._operations import utils


Expand Down
10 changes: 10 additions & 0 deletions packages/bigframes/bigframes/series.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,6 +353,16 @@ def _struct_fields(self) -> List[str]:
struct_type = typing.cast(pa.StructType, self._dtype.pyarrow_dtype)
return [struct_type.field(i).name for i in range(struct_type.num_fields)]

@property
def sql(self) -> str:
"""Compiles this Series's expression tree to SQL.

Returns:
A string representing the compiled SQL.
"""

return self.to_frame().sql

@validations.requires_ordering()
def transpose(self) -> Series:
return self
Expand Down
6 changes: 6 additions & 0 deletions packages/bigframes/tests/system/small/test_series.py
Original file line number Diff line number Diff line change
Expand Up @@ -4999,3 +4999,9 @@ def test_series_item_with_empty(session):

with pytest.raises(ValueError, match=re.escape(expected_message)):
bf_s_empty.item()


def test_series_sql(session):
s = bigframes.pandas.Series([], session=session)

assert len(s.sql) > 0
Comment thread
sycai marked this conversation as resolved.
Loading