add type annotation

datahub-project · May 23, 2022 · fb32b07 · fb32b07
1 parent 7037b3f
commit fb32b07
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/bigquery.py b/metadata-ingestion/src/datahub/ingestion/source/sql/bigquery.py
@@ -696,7 +696,9 @@ def get_extra_tags(
         self, inspector: Inspector, schema: str, table: str
     ) -> Dict[str, List[str]]:
         extra_tags: Dict[str, List[str]] = {}
-        partition = self.get_latest_partition(schema, table)
+        partition: Optional[BigQueryPartitionColumn] = self.get_latest_partition(
+            schema, table
+        )
         if partition:
             extra_tags[partition.column_name] = [Constants.TAG_PARTITION_KEY]
         return extra_tags