Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit aa1ddf4

Browse files
committed
chore: include pandas accessors in api logging
1 parent 1126cec commit aa1ddf4

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

bigframes/extensions/pandas/dataframe_accessor.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,12 @@
1818
import pandas.api.extensions
1919

2020
import bigframes.core.global_session as bf_session
21+
from bigframes.core.logging import log_adapter
2122
import bigframes.pandas as bpd
2223

2324

24-
class AIAccessor:
25+
@log_adapter.class_logger
26+
class PandasAIAccessor:
2527
"""
2628
Pandas DataFrame accessor for BigQuery AI functions.
2729
"""
@@ -101,7 +103,8 @@ def forecast(
101103

102104

103105
@pandas.api.extensions.register_dataframe_accessor("bigquery")
104-
class BigQueryDataFrameAccessor:
106+
@log_adapter.class_logger
107+
class PandasBigQueryDataFrameAccessor:
105108
"""
106109
Pandas DataFrame accessor for BigQuery DataFrames functionality.
107110
@@ -112,11 +115,11 @@ def __init__(self, pandas_obj: pandas.DataFrame):
112115
self._obj = pandas_obj
113116

114117
@property
115-
def ai(self) -> "AIAccessor":
118+
def ai(self) -> "PandasAIAccessor":
116119
"""
117120
Accessor for BigQuery AI functions.
118121
"""
119-
return AIAccessor(self._obj)
122+
return PandasAIAccessor(self._obj)
120123

121124
def sql_scalar(self, sql_template: str, *, output_dtype=None, session=None):
122125
"""

0 commit comments

Comments
 (0)