From 63960350edf6d88d680c7a77b6972313492f2541 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tim=20Swe=C3=B1a?= Date: Fri, 1 May 2026 20:15:39 +0000 Subject: [PATCH 1/2] fix: avoid `copy` argument warning in `to_pandas` --- packages/bigframes/bigframes/core/blocks.py | 4 ++-- packages/bigframes/bigframes/dataframe.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/packages/bigframes/bigframes/core/blocks.py b/packages/bigframes/bigframes/core/blocks.py index 5de65dcb61a2..b086b0860b08 100644 --- a/packages/bigframes/bigframes/core/blocks.py +++ b/packages/bigframes/bigframes/core/blocks.py @@ -847,8 +847,8 @@ def _materialize_local( ) else: raw_df = result_batches.to_pandas() - df = self._copy_index_to_pandas(raw_df) - df.set_axis(self.column_labels, axis=1, copy=False) + + df = self._copy_index_to_pandas(raw_df).set_axis(self.column_labels, axis=1) return df, execute_result.query_job def split( diff --git a/packages/bigframes/bigframes/dataframe.py b/packages/bigframes/bigframes/dataframe.py index c4b321c66880..424c7614d631 100644 --- a/packages/bigframes/bigframes/dataframe.py +++ b/packages/bigframes/bigframes/dataframe.py @@ -1743,7 +1743,7 @@ def to_pandas( ) if query_job: self._set_internal_query_job(query_job) - return df.set_axis(self._block.column_labels, axis=1, copy=False) + return df.set_axis(self._block.column_labels, axis=1) def to_pandas_batches( self, @@ -1869,7 +1869,7 @@ def peek( raise ValueError( "Cannot peek efficiently when data has aggregates, joins or window functions applied. Use force=True to fully compute dataframe." ) - return maybe_result.set_axis(self._block.column_labels, axis=1, copy=False) + return maybe_result.set_axis(self._block.column_labels, axis=1) def nlargest( self, From 76a1fdaeac1dfc8a228da2fa976024fc10f05e06 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tim=20Swe=C3=B1a=20=28Swast=29?= Date: Mon, 4 May 2026 15:26:04 -0500 Subject: [PATCH 2/2] Apply suggestions from code review Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- packages/bigframes/bigframes/core/blocks.py | 4 ++-- packages/bigframes/bigframes/dataframe.py | 6 ++++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/packages/bigframes/bigframes/core/blocks.py b/packages/bigframes/bigframes/core/blocks.py index b086b0860b08..b9a246fc0360 100644 --- a/packages/bigframes/bigframes/core/blocks.py +++ b/packages/bigframes/bigframes/core/blocks.py @@ -847,8 +847,8 @@ def _materialize_local( ) else: raw_df = result_batches.to_pandas() - - df = self._copy_index_to_pandas(raw_df).set_axis(self.column_labels, axis=1) + df = self._copy_index_to_pandas(raw_df) + df.columns = self.column_labels return df, execute_result.query_job def split( diff --git a/packages/bigframes/bigframes/dataframe.py b/packages/bigframes/bigframes/dataframe.py index 424c7614d631..3f3dfb5e0c16 100644 --- a/packages/bigframes/bigframes/dataframe.py +++ b/packages/bigframes/bigframes/dataframe.py @@ -1743,7 +1743,8 @@ def to_pandas( ) if query_job: self._set_internal_query_job(query_job) - return df.set_axis(self._block.column_labels, axis=1) + df.columns = self._block.column_labels + return df def to_pandas_batches( self, @@ -1869,7 +1870,8 @@ def peek( raise ValueError( "Cannot peek efficiently when data has aggregates, joins or window functions applied. Use force=True to fully compute dataframe." ) - return maybe_result.set_axis(self._block.column_labels, axis=1) + maybe_result.columns = self._block.column_labels + return maybe_result def nlargest( self,