feat: Add More Specific Type Annotations for Row Dictionaries (#1295)

The keys must be strings as they represent column names. Update type annotations to reflect this. Co-authored-by: aribray <[email protected]>
googleapis · Nov 16, 2022 · eb49873 · eb49873
1 parent 207aa50
commit eb49873
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/google/cloud/bigquery/client.py b/google/cloud/bigquery/client.py
@@ -3349,10 +3349,10 @@ def query(
     def insert_rows(
         self,
         table: Union[Table, TableReference, str],
-        rows: Union[Iterable[Tuple], Iterable[Dict]],
+        rows: Union[Iterable[Tuple], Iterable[Mapping[str, Any]]],
         selected_fields: Sequence[SchemaField] = None,
         **kwargs,
-    ) -> Sequence[dict]:
+    ) -> Sequence[Dict[str, Any]]:
         """Insert rows into a table via the streaming API.
 
         See
@@ -3470,7 +3470,7 @@ def insert_rows_from_dataframe(
     def insert_rows_json(
         self,
         table: Union[Table, TableReference, TableListItem, str],
-        json_rows: Sequence[Dict],
+        json_rows: Sequence[Mapping[str, Any]],
         row_ids: Union[
             Iterable[Optional[str]], AutoRowIDs, None
         ] = AutoRowIDs.GENERATE_UUID,