lint

datahub-project · Oct 6, 2023 · 37b207b · 37b207b
1 parent ab6869e
commit 37b207b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/unity/proxy.py b/metadata-ingestion/src/datahub/ingestion/source/unity/proxy.py
@@ -3,7 +3,6 @@
 """
 import dataclasses
 import logging
-from concurrent.futures import ThreadPoolExecutor, wait
 from datetime import datetime, timezone
 from typing import Any, Dict, Iterable, List, Optional, Union
 from unittest.mock import patch

diff --git a/metadata-ingestion/src/datahub/ingestion/source/unity/source.py b/metadata-ingestion/src/datahub/ingestion/source/unity/source.py
@@ -1,7 +1,7 @@
 import logging
 import re
 import time
-from concurrent.futures import ThreadPoolExecutor, wait
+from concurrent.futures import ThreadPoolExecutor
 from datetime import timedelta
 from typing import Dict, Iterable, List, Optional, Set, Union
 from urllib.parse import urljoin
@@ -394,7 +394,7 @@ def process_table(self, table: Table, schema: Schema) -> Iterable[MetadataWorkUn
             )
         ]
 
-    def ingest_lineage(self, table: Table) -> Optional[MetadataWorkUnit]:
+    def ingest_lineage(self, table: Table) -> Optional[UpstreamLineageClass]:
         if self.config.include_table_lineage:
             self.unity_catalog_api_proxy.table_lineage(
                 table, include_entity_lineage=self.config.include_notebooks