fix

Author: GarrettWu

bigframes/bigquery/_operations/ai.py

@@ -1018,7 +1018,7 @@ def _convert_series(
 
 def _resolve_connection_id(series: series.Series, connection_id: str | None):
     return clients.get_canonical_bq_connection_id(
-        connection_id or series._session._bq_connection,
+        connection_id or series._session.bq_connection,
         series._session._project,
         series._session._location,
     )

bigframes/core/blocks.py

@@ -442,7 +442,7 @@ def reset_index(
             level_ids = self.index_columns
 
         expr = self._expr
-        replacement_idx_type = replacement or self.session._default_index_type  # type: ignore
+        replacement_idx_type = replacement or self.session._default_index_type
         if set(self.index_columns) > set(level_ids):
             new_index_cols = [col for col in self.index_columns if col not in level_ids]
             new_index_labels = [self.col_id_to_index_name[id] for id in new_index_cols]
@@ -2382,7 +2382,7 @@ def merge(
         if (
             self.index.is_null
             or other.index.is_null
-            or self.session._default_index_type == bigframes.enums.DefaultIndexKind.NULL  # type: ignore
+            or self.session._default_index_type == bigframes.enums.DefaultIndexKind.NULL
         ):
             return Block(joined_expr, index_columns=[], column_labels=labels)
         elif index_cols:

bigframes/dataframe.py

@@ -4233,7 +4233,7 @@ def to_gbq(
 
             # The client code owns this table reference now
             temp_table_ref = (
-                self._session._anon_dataset_manager.generate_unique_resource_id()  # type: ignore
+                self._session._anon_dataset_manager.generate_unique_resource_id()
             )
             destination_table = f"{temp_table_ref.project}.{temp_table_ref.dataset_id}.{temp_table_ref.table_id}"
 

bigframes/functions/_function_session.py

@@ -162,7 +162,7 @@ def _resolve_bigquery_connection_id(
     ) -> str:
         """Resolves BigQuery connection id."""
         if not bigquery_connection:
-            bigquery_connection = session._bq_connection  # type: ignore
+            bigquery_connection = session.bq_connection  # type: ignore
 
         bigquery_connection = clients.get_canonical_bq_connection_id(
             bigquery_connection,

bigframes/operations/blob.py

@@ -311,7 +311,7 @@ def _resolve_connection(self, connection: Optional[str] = None) -> str:
         Raises:
             ValueError: If the connection cannot be resolved to a valid string.
         """
-        connection = connection or self._data._block.session._bq_connection
+        connection = connection or self._data._block.session.bq_connection
         return clients.get_canonical_bq_connection_id(
             connection,
             default_project=self._data._block.session._project,

bigframes/session/__init__.py

@@ -375,7 +375,7 @@ def _anonymous_dataset(self):
         return self._anon_dataset_manager.dataset
 
     @property
-    def _bq_connection(self) -> str:
+    def bq_connection(self) -> str:
         msg = bfe.format_message(
             f"""You are using the BigFrames session default connection: {self._bq_connection},
             which can be different from the BigQuery project default connection.
@@ -384,10 +384,6 @@ def _bq_connection(self) -> str:
         warnings.warn(msg, category=FutureWarning)
         return self._bq_connection
 
-    @_bq_connection.setter
-    def _bq_connection(self, value):
-        self._bq_connection = value
-
     def __hash__(self):
         # Stable hash needed to use in expression tree
         return hash(str(self._session_id))
@@ -2267,7 +2263,7 @@ def _create_bq_connection(
     ) -> str:
         """Create the connection with the session settings and try to attach iam role to the connection SA.
         If any of project, location or connection isn't specified, use the session defaults. Returns fully-qualified connection name."""
-        connection = self._bq_connection if not connection else connection
+        connection = self.bq_connection if not connection else connection
         connection = bigframes.clients.get_canonical_bq_connection_id(
             connection_id=connection,
             default_project=self._project,