fix(ingest/dagster): Fix JobSnapshot import is broken (datahub-projec…

…t#11672)
acryldata · Oct 22, 2024 · 445e543 · 445e543
1 parent 7b82db2
commit 445e543
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...a-ingestion-modules/dagster-plugin/src/datahub_dagster_plugin/client/dagster_generator.py b/...a-ingestion-modules/dagster-plugin/src/datahub_dagster_plugin/client/dagster_generator.py
@@ -12,7 +12,13 @@
     TableSchemaMetadataValue,
 )
 from dagster._core.execution.stats import RunStepKeyStatsSnapshot, StepEventStatus
-from dagster._core.snap import JobSnapshot
+
+try:
+    from dagster._core.snap import JobSnapshot  # type: ignore[attr-defined]
+except ImportError:
+    # Import changed since Dagster 1.8.12  to this -> https://github.com/dagster-io/dagster/commit/29a37d1f0260cfd112849633d1096ffc916d6c95
+    from dagster._core.snap import JobSnap as JobSnapshot
+
 from dagster._core.snap.node import OpDefSnap
 from dagster._core.storage.dagster_run import DagsterRun, DagsterRunStatsSnapshot
 from datahub.api.entities.datajob import DataFlow, DataJob