Export AutomationContext and AutomationResult

dagster-io · Oct 11, 2024 · 8acc87b · 8acc87b
1 parent 9be9003
commit 8acc87b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/python_modules/dagster/dagster/__init__.py b/python_modules/dagster/dagster/__init__.py
@@ -165,10 +165,14 @@
 )
 from dagster._core.definitions.declarative_automation.automation_condition import (
     AutomationCondition as AutomationCondition,
+    AutomationResult as AutomationResult,
 )
 from dagster._core.definitions.declarative_automation.automation_condition_tester import (
     evaluate_automation_conditions as evaluate_automation_conditions,
 )
+from dagster._core.definitions.declarative_automation.automation_context import (
+    AutomationContext as AutomationContext,
+)
 from dagster._core.definitions.decorators.asset_check_decorator import (
     asset_check as asset_check,
     multi_asset_check as multi_asset_check,

diff --git a/...n_tests/automation_condition_tests/user_defined/test_user_defined_automation_condition.py b/...n_tests/automation_condition_tests/user_defined/test_user_defined_automation_condition.py
@@ -1,8 +1,13 @@
 import logging
 
-from dagster import AutomationCondition, DagsterInstance, asset, evaluate_automation_conditions
-from dagster._core.definitions.declarative_automation.automation_condition import AutomationResult
-from dagster._core.definitions.declarative_automation.automation_context import AutomationContext
+from dagster import (
+    AutomationCondition,
+    AutomationContext,
+    AutomationResult,
+    DagsterInstance,
+    asset,
+    evaluate_automation_conditions,
+)
 
 
 def test_cursoring() -> None:

diff --git a/...itions_tests/declarative_automation_tests/scenario_utils/automation_condition_scenario.py b/...itions_tests/declarative_automation_tests/scenario_utils/automation_condition_scenario.py
@@ -5,7 +5,7 @@
 
 import dagster._check as check
 import mock
-from dagster import AssetKey
+from dagster import AssetKey, AutomationContext
 from dagster._core.asset_graph_view.asset_graph_view import AssetGraphView
 from dagster._core.definitions.asset_daemon_cursor import AssetDaemonCursor
 from dagster._core.definitions.auto_materialize_policy import AutoMaterializePolicy
@@ -16,7 +16,6 @@
 from dagster._core.definitions.declarative_automation.automation_condition_evaluator import (
     AutomationConditionEvaluator,
 )
-from dagster._core.definitions.declarative_automation.automation_context import AutomationContext
 from dagster._core.definitions.declarative_automation.operators.boolean_operators import (
     AndAutomationCondition,
 )