fix test issues

Author: TrevorBergeron

bigframes/functions/_function_client.py

@@ -586,7 +586,7 @@ def provision_bq_remote_function(
         cloud_function_vpc_connector_egress_settings: str | None,
         cloud_function_memory_mib: int | None,
         cloud_function_cpus: float | None,
-        cloud_function_ingress_settings: str | None,
+        cloud_function_ingress_settings: str,
     ):
         """Provision a BigQuery remote function."""
         # Augment user package requirements with any internal package
@@ -620,7 +620,7 @@ def provision_bq_remote_function(
             or "private-ranges-only",
             memory_mib=cloud_function_memory_mib,
             cpus=cloud_function_cpus,
-            ingress_settings=cloud_function_ingress_settings or "internal_only",
+            ingress_settings=cloud_function_ingress_settings,
             workers=workers,
             threads=threads,
             concurrency=concurrency,

bigframes/functions/udf_def.py

@@ -113,8 +113,7 @@ def bf_type(self) -> bigframes.dtypes.Dtype:
     @property
     def sql_type(self) -> str:
         sdk_type = function_typing.sdk_type_from_python_type(self._py_type)
-        assert sdk_type.type_kind is not None
-        return sdk_type.type_kind.name
+        return function_typing.sdk_type_to_sql_string(sdk_type)
 
     def stable_hash(self) -> bytes:
         hash_val = hashlib.md5()
@@ -229,13 +228,15 @@ def to_sql_input_signature(self) -> str:
         )
 
     @property
-    def protocol_metadata(self) -> str:
+    def protocol_metadata(self) -> str | None:
         import bigframes.functions._utils
 
         # TODO: The output field itself should handle this, to handle protocol versioning.
-        return bigframes.functions._utils.get_bigframes_metadata(
-            python_output_type=self.output.py_type
-        )
+        if isinstance(self.output, VirtualListTypeV1):
+            return bigframes.functions._utils.get_bigframes_metadata(
+                python_output_type=self.output.py_type
+            )
+        return None
 
     @property
     def is_virtual(self) -> bool:

tests/system/large/ml/test_ensemble.py

@@ -155,7 +155,7 @@ def test_xgbclassifier_default_params(penguins_df_default_index, dataset_id):
     )
 
 
-# @pytest.mark.flaky(retries=2)
+@pytest.mark.flaky(retries=2)
 def test_xgbclassifier_dart_booster_multiple_params(
     penguins_df_default_index, dataset_id
 ):

tests/system/large/ml/test_llm.py

@@ -63,7 +63,7 @@ def test_create_load_gemini_text_generator_model(
         "gemini-2.5-flash-lite",
     ),
 )
-# @pytest.mark.flaky(retries=2)
+@pytest.mark.flaky(retries=2)
 def test_gemini_text_generator_predict_default_params_success(
     llm_text_df, model_name, session, bq_connection
 ):

tests/system/small/functions/test_remote_function.py

@@ -106,7 +106,7 @@ def get_bq_connection_id_path_format(connection_id_dot_format):
     return f"projects/{fields[0]}/locations/{fields[1]}/connections/{fields[2]}"
 
 
-# @pytest.mark.flaky(retries=2, delay=120)
+@pytest.mark.flaky(retries=2, delay=120)
 def test_remote_function_direct_no_session_param(
     bigquery_client,
     bigqueryconnection_client,