feat: add type hint for public methods

.gitignore

@@ -29,6 +29,7 @@ pip-log.txt
 .nox
 .cache
 .pytest_cache
+.pytype
 
 
 # Mac

google/cloud/bigquery/_helpers.py

@@ -20,12 +20,12 @@
 import re
 import six
 
-from google.cloud._helpers import UTC
-from google.cloud._helpers import _date_from_iso8601_date
-from google.cloud._helpers import _datetime_from_microseconds
-from google.cloud._helpers import _RFC3339_MICROS
-from google.cloud._helpers import _RFC3339_NO_FRACTION
-from google.cloud._helpers import _to_bytes
+from google.cloud._helpers import UTC  # type: ignore
+from google.cloud._helpers import _date_from_iso8601_date  # type: ignore
+from google.cloud._helpers import _datetime_from_microseconds  # type: ignore
+from google.cloud._helpers import _RFC3339_MICROS  # type: ignore
+from google.cloud._helpers import _RFC3339_NO_FRACTION  # type: ignore
+from google.cloud._helpers import _to_bytes  # type: ignore
 
 _RFC3339_MICROS_NO_ZULU = "%Y-%m-%dT%H:%M:%S.%f"
 _TIMEONLY_WO_MICROS = "%H:%M:%S"

google/cloud/bigquery/_http.py

@@ -14,7 +14,7 @@
 
 """Create / interact with Google BigQuery connections."""
 
-from google.cloud import _http
+from google.cloud import _http  # type: ignore
 
 from google.cloud.bigquery import __version__
 

google/cloud/bigquery/_pandas_helpers.py

@@ -23,13 +23,13 @@
 from six.moves import queue
 
 try:
-    import pandas
+    import pandas  # type: ignore
 except ImportError:  # pragma: NO COVER
     pandas = None
 
 try:
-    import pyarrow
-    import pyarrow.parquet
+    import pyarrow  # type: ignore
+    import pyarrow.parquet  # type: ignore
 except ImportError:  # pragma: NO COVER
     pyarrow = None
 
@@ -350,6 +350,7 @@ def augment_schema(dataframe, current_bq_schema):
     Returns:
         Optional[Sequence[google.cloud.bigquery.schema.SchemaField]]
     """
+    # pytype: disable=attribute-error
     augmented_schema = []
     unknown_type_fields = []
 
@@ -383,6 +384,7 @@ def augment_schema(dataframe, current_bq_schema):
         return None
 
     return augmented_schema
+    # pytype: enable=attribute-error
 
 
 def dataframe_to_arrow(dataframe, bq_schema):
@@ -600,7 +602,7 @@ def _download_table_bqstorage(
 
     # Passing a BQ Storage client in implies that the BigQuery Storage library
     # is available and can be imported.
-    from google.cloud import bigquery_storage
+    from google.cloud import bigquery_storage  # type: ignore
 
     if "$" in table.table_id:
         raise ValueError(

google/cloud/bigquery/_tqdm_helpers.py

@@ -19,7 +19,7 @@
 import warnings
 
 try:
-    import tqdm
+    import tqdm  # type: ignore
 except ImportError:  # pragma: NO COVER
     tqdm = None