Skip to content

Commit f3061f8

Browse files
authored
chore: filter internal calls. (#1329)
1 parent 5c2a2c6 commit f3061f8

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

bigframes/core/log_adapter.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,9 @@ def submit_pandas_labels(
6363
- 'PANDAS_PARAM_TRACKING_TASK': Indicates that the unimplemented feature is a
6464
parameter of a method.
6565
"""
66+
if method_name.startswith("_") and not method_name.startswith("__"):
67+
return
68+
6669
labels_dict = {
6770
"task": task,
6871
"class_name": class_name.lower(),

tests/unit/core/test_log_adapter.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,3 +155,13 @@ def test_submit_pandas_labels_without_valid_params_for_param_logging(mock_bqclie
155155

156156
# For param tracking task without kwargs, we won't submit labels
157157
mock_bqclient.query.assert_not_called()
158+
159+
160+
def test_submit_pandas_labels_with_internal_method(mock_bqclient):
161+
log_adapter.submit_pandas_labels(
162+
mock_bqclient,
163+
"Series",
164+
"_repr_latex_",
165+
task=log_adapter.PANDAS_API_TRACKING_TASK,
166+
)
167+
mock_bqclient.query.assert_not_called()

0 commit comments

Comments
 (0)