From 7793a8ee568fcb2b4e8b33517e9e727b19adc07a Mon Sep 17 00:00:00 2001 From: Shuowei Li Date: Wed, 17 Jun 2026 21:59:08 +0000 Subject: [PATCH 1/3] tests: skip test_remote_function_connection_w_location (b/525124882) --- .../tests/system/small/functions/test_remote_function.py | 1 + 1 file changed, 1 insertion(+) diff --git a/packages/bigframes/tests/system/small/functions/test_remote_function.py b/packages/bigframes/tests/system/small/functions/test_remote_function.py index 869b26ca38c5..0d58a5619d6f 100644 --- a/packages/bigframes/tests/system/small/functions/test_remote_function.py +++ b/packages/bigframes/tests/system/small/functions/test_remote_function.py @@ -168,6 +168,7 @@ def square(x): assert_frame_equal(bf_result, pd_result) +@pytest.mark.skip(reason="b/525124882: connection with location test failing") @pytest.mark.flaky(retries=2, delay=120) def test_remote_function_connection_w_location( session, From 6e3a5d12e69b1208385bdca98c17e38705219ae1 Mon Sep 17 00:00:00 2001 From: Shuowei Li Date: Wed, 17 Jun 2026 23:06:43 +0000 Subject: [PATCH 2/3] fix(functions): handle empty cloud function endpoints --- packages/bigframes/bigframes/functions/_function_session.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/bigframes/bigframes/functions/_function_session.py b/packages/bigframes/bigframes/functions/_function_session.py index e369b0b39bfd..213ac6638490 100644 --- a/packages/bigframes/bigframes/functions/_function_session.py +++ b/packages/bigframes/bigframes/functions/_function_session.py @@ -592,7 +592,9 @@ def wrapper(func): if reuse is not None: cf_endpoint = self._function_client.get_cloud_function_endpoint(cf_name) - if cf_endpoint is None: + # If the endpoint is empty, the function might exist but the URL propagation is pending. + # Running create_cloud_function will handle AlreadyExists and retry endpoint fetching. + if not cf_endpoint: cf_endpoint = self._function_client.create_cloud_function( cf_name, cloud_func_spec ) From dd3496107e8b1134872ff56e5b718b048c177cff Mon Sep 17 00:00:00 2001 From: Shuowei Li Date: Wed, 17 Jun 2026 23:08:00 +0000 Subject: [PATCH 3/3] Revert "tests: skip test_remote_function_connection_w_location (b/525124882)" This reverts commit 7793a8ee568fcb2b4e8b33517e9e727b19adc07a. --- .../tests/system/small/functions/test_remote_function.py | 1 - 1 file changed, 1 deletion(-) diff --git a/packages/bigframes/tests/system/small/functions/test_remote_function.py b/packages/bigframes/tests/system/small/functions/test_remote_function.py index 0d58a5619d6f..869b26ca38c5 100644 --- a/packages/bigframes/tests/system/small/functions/test_remote_function.py +++ b/packages/bigframes/tests/system/small/functions/test_remote_function.py @@ -168,7 +168,6 @@ def square(x): assert_frame_equal(bf_result, pd_result) -@pytest.mark.skip(reason="b/525124882: connection with location test failing") @pytest.mark.flaky(retries=2, delay=120) def test_remote_function_connection_w_location( session,