From 0c8a07f7ea64a99d1e0353a0e3f5f1172a090068 Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Wed, 26 Mar 2025 01:22:42 +0000 Subject: [PATCH] chore: add logging for blob operations --- bigframes/operations/blob.py | 2 ++ bigframes/session/__init__.py | 8 ++++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/bigframes/operations/blob.py b/bigframes/operations/blob.py index a37615efbb..7c0b6d4d1c 100644 --- a/bigframes/operations/blob.py +++ b/bigframes/operations/blob.py @@ -23,6 +23,7 @@ import requests from bigframes import clients +from bigframes.core import log_adapter import bigframes.dataframe import bigframes.exceptions as bfe from bigframes.operations import base @@ -33,6 +34,7 @@ FILE_EXT_REGEX = r"(\.[0-9a-zA-Z]+$)" +@log_adapter.class_logger class BlobAccessor(base.SeriesMethods): def __init__(self, *args, **kwargs): if not bigframes.options.experiments.blob: diff --git a/bigframes/session/__init__.py b/bigframes/session/__init__.py index dfee41c90b..26db6c2f34 100644 --- a/bigframes/session/__init__.py +++ b/bigframes/session/__init__.py @@ -1757,7 +1757,9 @@ def from_glob_path( table = self._create_object_table(path, connection) - s = self.read_gbq(table)["uri"].str.to_blob(connection) + s = self._loader.read_gbq_table(table, api_name="from_glob_path")[ + "uri" + ].str.to_blob(connection) return s.rename(name).to_frame() def _create_bq_connection( @@ -1807,7 +1809,9 @@ def read_gbq_object_table( table = self.bqclient.get_table(object_table) connection = table._properties["externalDataConfiguration"]["connectionId"] - s = self.read_gbq(object_table)["uri"].str.to_blob(connection) + s = self._loader.read_gbq_table(object_table, api_name="read_gbq_object_table")[ + "uri" + ].str.to_blob(connection) return s.rename(name).to_frame()