diff --git a/.librarian/generator-input/client-post-processing/integrate-isolated-handwritten-code.yaml b/.librarian/generator-input/client-post-processing/integrate-isolated-handwritten-code.yaml index e3161120ec0a..7ee33fd85053 100644 --- a/.librarian/generator-input/client-post-processing/integrate-isolated-handwritten-code.yaml +++ b/.librarian/generator-input/client-post-processing/integrate-isolated-handwritten-code.yaml @@ -290,3 +290,14 @@ replacements: after: | UNIT_TEST_EXTRAS: List[str] = ["pandas", "storage"] count: 1 + - paths: [ + packages/google-cloud-automl/noxfile.py, + ] + before: | + "types-protobuf", + \ \) + after: | + "types-protobuf", + "pandas-stubs", + ) + count: 1 diff --git a/packages/google-cloud-automl/google/cloud/automl_v1beta1/services/tables/gcs_client.py b/packages/google-cloud-automl/google/cloud/automl_v1beta1/services/tables/gcs_client.py index 99d40da2867c..3fa86d22229b 100644 --- a/packages/google-cloud-automl/google/cloud/automl_v1beta1/services/tables/gcs_client.py +++ b/packages/google-cloud-automl/google/cloud/automl_v1beta1/services/tables/gcs_client.py @@ -24,12 +24,14 @@ try: import pandas except ImportError: # pragma: NO COVER - pandas = None + pandas = None # type: ignore[assignment] try: - from google.cloud import storage + # TODO(https://github.com/googleapis/python-storage/issues/318): + # Remove `type: ignore` once this bug is fixed + from google.cloud import storage # type: ignore[attr-defined] except ImportError: # pragma: NO COVER - storage = None + storage = None # type: ignore[assignment] _LOGGER = logging.getLogger(__name__) _PANDAS_REQUIRED = "pandas is required to verify type DataFrame." diff --git a/packages/google-cloud-automl/noxfile.py b/packages/google-cloud-automl/noxfile.py index bcafe0750c9a..4a482fbfa495 100644 --- a/packages/google-cloud-automl/noxfile.py +++ b/packages/google-cloud-automl/noxfile.py @@ -101,6 +101,7 @@ def mypy(session): "mypy<1.16.0", "types-requests", "types-protobuf", + "pandas-stubs", ) session.install(".") session.run(