Sync with main (squashed)

Author: sycai

.librarian/state.yaml

@@ -71,8 +71,13 @@ case ${TEST_TYPE} in
     unit)
         case ${PY_VERSION} in
         "3.9")
-            nox -s unit-3.9
-            retval=$?
+            if nox --list-sessions | grep -q "unit-3.9"; then
+                nox -s unit-3.9
+                retval=$?
+            else
+                echo "Skipping unit-3.9 as it is not supported by this package."
+                retval=0
+            fi
             ;;
         "3.10")
             nox -s unit-3.10

ci/run_single_test.sh

@@ -329,7 +329,7 @@ def test_ai_if(session):
         max_error_ratio=0.5,
     )
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(s1)
     assert result.dtype == dtypes.BOOL_DTYPE
 
 
@@ -342,7 +342,7 @@ def test_ai_if_multi_model(session, bq_connection):
 
     result = bbq.ai.if_((df["image"], " contains an animal"))
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(df)
     assert result.dtype == dtypes.BOOL_DTYPE
 
 
@@ -351,7 +351,7 @@ def test_ai_classify(session):
 
     result = bbq.ai.classify(s, ["animal", "plant"])
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(s)
     assert result.dtype == dtypes.STRING_DTYPE
 
 
@@ -364,7 +364,7 @@ def test_ai_classify_multi_model(session, bq_connection):
 
     result = bbq.ai.classify(df["image"], ["photo", "cartoon"])
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(df)
     assert result.dtype == dtypes.STRING_DTYPE
 
 
@@ -374,7 +374,7 @@ def test_ai_score(session):
 
     result = bbq.ai.score(prompt)
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(s)
     assert result.dtype == dtypes.FLOAT_DTYPE
 
 
@@ -386,7 +386,7 @@ def test_ai_score_multi_model(session):
 
     result = bbq.ai.score(prompt)
 
-    assert _contains_no_nulls(result)
+    assert len(result) == len(df)
     assert result.dtype == dtypes.FLOAT_DTYPE
 
 

packages/bigframes/tests/system/small/bigquery/test_ai.py

@@ -28,7 +28,6 @@
 DEFAULT_PYTHON_VERSION = "3.14"
 
 UNIT_TEST_PYTHON_VERSIONS = [
-    "3.9",
     "3.10",
     "3.11",
     "3.12",
@@ -123,8 +122,7 @@ def default(session, django_version="5.2"):
 @nox.session(python=ALL_PYTHON)
 def unit(session):
     """Run the unit test suite."""
-    if session.python == "3.9":
-        session.skip("Python 3.9 is not supported for Django 5.2 tests")
+    # TODO: Remove this check once support for Python 3.14 is added to Protobuf
     if session.python == "3.14":
         session.skip("Protobuf upb implementation is not supported in Python 3.14 yet")
     print("Unit tests with django 5.2")

packages/django-google-spanner/noxfile.py

@@ -73,7 +73,7 @@ def gapic_generator_python():
         strip_prefix = "rules_gapic-%s" % _rules_gapic_version,
         urls = ["https://github.com/googleapis/rules_gapic/archive/v%s.tar.gz" % _rules_gapic_version],
     )
-    _commit_sha = "fae3e6e091418d6343902debaf545cfc8f32c3ff"
+    _commit_sha = "2638026851b9110c280d75338581a752a4271340"
     _maybe(
         http_archive,
         name = "com_google_googleapis",

packages/gapic-generator/repositories.bzl

@@ -21,6 +21,7 @@
 from google.cloud.asset_v1.services.asset_service.client import AssetServiceClient
 from google.cloud.asset_v1.services.asset_service.async_client import AssetServiceAsyncClient
 
+from google.cloud.asset_v1.types.asset_enrichment_resourceowners import ResourceOwners
 from google.cloud.asset_v1.types.asset_service import AnalyzeIamPolicyLongrunningMetadata
 from google.cloud.asset_v1.types.asset_service import AnalyzeIamPolicyLongrunningRequest
 from google.cloud.asset_v1.types.asset_service import AnalyzeIamPolicyLongrunningResponse
@@ -83,8 +84,10 @@
 from google.cloud.asset_v1.types.asset_service import UpdateSavedQueryRequest
 from google.cloud.asset_v1.types.asset_service import ContentType
 from google.cloud.asset_v1.types.assets import Asset
+from google.cloud.asset_v1.types.assets import AssetEnrichment
 from google.cloud.asset_v1.types.assets import AttachedResource
 from google.cloud.asset_v1.types.assets import ConditionEvaluation
+from google.cloud.asset_v1.types.assets import EffectiveTagDetails
 from google.cloud.asset_v1.types.assets import IamPolicyAnalysisResult
 from google.cloud.asset_v1.types.assets import IamPolicyAnalysisState
 from google.cloud.asset_v1.types.assets import IamPolicySearchResult
@@ -95,12 +98,14 @@
 from google.cloud.asset_v1.types.assets import RelationshipAttributes
 from google.cloud.asset_v1.types.assets import Resource
 from google.cloud.asset_v1.types.assets import ResourceSearchResult
+from google.cloud.asset_v1.types.assets import Tag
 from google.cloud.asset_v1.types.assets import TemporalAsset
 from google.cloud.asset_v1.types.assets import TimeWindow
 from google.cloud.asset_v1.types.assets import VersionedResource
 
 __all__ = ('AssetServiceClient',
     'AssetServiceAsyncClient',
+    'ResourceOwners',
     'AnalyzeIamPolicyLongrunningMetadata',
     'AnalyzeIamPolicyLongrunningRequest',
     'AnalyzeIamPolicyLongrunningResponse',
@@ -163,8 +168,10 @@
     'UpdateSavedQueryRequest',
     'ContentType',
     'Asset',
+    'AssetEnrichment',
     'AttachedResource',
     'ConditionEvaluation',
+    'EffectiveTagDetails',
     'IamPolicyAnalysisResult',
     'IamPolicyAnalysisState',
     'IamPolicySearchResult',
@@ -175,6 +182,7 @@
     'RelationshipAttributes',
     'Resource',
     'ResourceSearchResult',
+    'Tag',
     'TemporalAsset',
     'TimeWindow',
     'VersionedResource',

packages/gapic-generator/tests/integration/goldens/asset/google/cloud/asset/__init__.py

@@ -26,6 +26,7 @@
 from .services.asset_service import AssetServiceClient
 from .services.asset_service import AssetServiceAsyncClient
 
+from .types.asset_enrichment_resourceowners import ResourceOwners
 from .types.asset_service import AnalyzeIamPolicyLongrunningMetadata
 from .types.asset_service import AnalyzeIamPolicyLongrunningRequest
 from .types.asset_service import AnalyzeIamPolicyLongrunningResponse
@@ -88,8 +89,10 @@
 from .types.asset_service import UpdateSavedQueryRequest
 from .types.asset_service import ContentType
 from .types.assets import Asset
+from .types.assets import AssetEnrichment
 from .types.assets import AttachedResource
 from .types.assets import ConditionEvaluation
+from .types.assets import EffectiveTagDetails
 from .types.assets import IamPolicyAnalysisResult
 from .types.assets import IamPolicyAnalysisState
 from .types.assets import IamPolicySearchResult
@@ -100,6 +103,7 @@
 from .types.assets import RelationshipAttributes
 from .types.assets import Resource
 from .types.assets import ResourceSearchResult
+from .types.assets import Tag
 from .types.assets import TemporalAsset
 from .types.assets import TimeWindow
 from .types.assets import VersionedResource
@@ -197,6 +201,7 @@ def _get_version(dependency_name):
 'AnalyzerOrgPolicy',
 'AnalyzerOrgPolicyConstraint',
 'Asset',
+'AssetEnrichment',
 'AssetServiceClient',
 'AttachedResource',
 'BatchGetAssetsHistoryRequest',
@@ -210,6 +215,7 @@ def _get_version(dependency_name):
 'CreateSavedQueryRequest',
 'DeleteFeedRequest',
 'DeleteSavedQueryRequest',
+'EffectiveTagDetails',
 'ExportAssetsRequest',
 'ExportAssetsResponse',
 'Feed',
@@ -246,6 +252,7 @@ def _get_version(dependency_name):
 'RelatedResources',
 'RelationshipAttributes',
 'Resource',
+'ResourceOwners',
 'ResourceSearchResult',
 'SavedQuery',
 'SearchAllIamPoliciesRequest',
@@ -254,6 +261,7 @@ def _get_version(dependency_name):
 'SearchAllResourcesResponse',
 'TableFieldSchema',
 'TableSchema',
+'Tag',
 'TemporalAsset',
 'TimeWindow',
 'UpdateFeedRequest',