Try to import new module for ExternalRepositoryOrigin.

Signed-off-by: Jakub Dardzinski <kuba0221@gmail.com>
OpenLineage · Mar 14, 2024 · fb773a5 · fb773a5
1 parent c3765bf
commit fb773a5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/integration/dagster/tests/conftest.py b/integration/dagster/tests/conftest.py
@@ -17,7 +17,11 @@
     EventLogRecord,
 )
 from dagster.core.execution.plan.objects import StepFailureData, StepSuccessData
-from dagster.core.host_representation import ExternalRepositoryOrigin
+
+try:
+    from dagster.core.host_representation import ExternalRepositoryOrigin
+except ImportError:
+    from dagster._core.remote_representation.origin import ExternalRepositoryOrigin
 from dagster.core.types.loadable_target_origin import LoadableTargetOrigin
 from dagster.version import __version__ as DAGSTER_VERSION