fix circular imports

datahub-project · Oct 13, 2022 · dc7c178 · dc7c178
1 parent 5c416aa
commit dc7c178
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/api/__init__.py b/metadata-ingestion/src/datahub/ingestion/api/__init__.py
@@ -1 +0,0 @@
-from datahub.ingestion.api.common import RecordEnvelope

diff --git a/metadata-ingestion/src/datahub/ingestion/api/common.py b/metadata-ingestion/src/datahub/ingestion/api/common.py
@@ -45,7 +45,7 @@ class PipelineContext:
     def __init__(
         self,
         run_id: str,
-        datahub_api: Optional[DatahubClientConfig] = None,
+        datahub_api: Optional["DatahubClientConfig"] = None,
         pipeline_name: Optional[str] = None,
         dry_run: bool = False,
         preview_mode: bool = False,

diff --git a/metadata-ingestion/src/datahub/ingestion/extractor/mce_extractor.py b/metadata-ingestion/src/datahub/ingestion/extractor/mce_extractor.py
@@ -3,7 +3,7 @@
 from datahub.configuration.common import ConfigModel
 from datahub.emitter.mce_builder import get_sys_time
 from datahub.emitter.mcp import MetadataChangeProposalWrapper
-from datahub.ingestion.api import RecordEnvelope
+from datahub.ingestion.api.common import RecordEnvelope
 from datahub.ingestion.api.source import Extractor, WorkUnit
 from datahub.ingestion.api.workunit import MetadataWorkUnit, UsageStatsWorkUnit
 from datahub.metadata.com.linkedin.pegasus2avro.mxe import (
Original file line number	Diff line number	Diff line change
		@@ -1 +0,0 @@
		from datahub.ingestion.api.common import RecordEnvelope