fix type annotation

dagster-io · Sep 20, 2023 · 5a064ff · 5a064ff
1 parent 160c957
commit 5a064ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python_modules/dagster-graphql/dagster_graphql_tests/graphql/repo.py b/python_modules/dagster-graphql/dagster_graphql_tests/graphql/repo.py
@@ -41,6 +41,7 @@
     Map,
     Noneable,
     Nothing,
+    OpExecutionContext,
     Out,
     Output,
     PythonObjectDagsterType,
@@ -733,7 +734,7 @@ def spawn() -> int:
     @op(
         required_resource_keys={"retry_count"},
     )
-    def fail(context: AssetExecutionContext, depth: int) -> int:
+    def fail(context: OpExecutionContext, depth: int) -> int:
         if context.resources.retry_count <= depth:
             raise Exception("fail")