fix type hints

Author: maximearmstrong

python_modules/libraries/dagster-dbt/dagster_dbt/cloud_v2/cli_invocation.py

@@ -1,7 +1,13 @@
 from collections.abc import Iterator, Mapping, Sequence
 from typing import Any, Optional, Union
 
-from dagster import AssetCheckEvaluation, AssetExecutionContext, AssetMaterialization, Output
+from dagster import (
+    AssetCheckEvaluation,
+    AssetCheckResult,
+    AssetExecutionContext,
+    AssetMaterialization,
+    Output,
+)
 from dagster._annotations import preview
 from dagster._record import record
 
@@ -46,7 +52,9 @@ def run(
             context=context,
         )
 
-    def wait(self) -> Iterator[Union[AssetMaterialization, AssetCheckEvaluation, Output]]:
+    def wait(
+        self,
+    ) -> Iterator[Union[AssetCheckEvaluation, AssetCheckResult, AssetMaterialization, Output]]:
         self.run_handler.wait_for_success()
         if "run_results.json" not in self.run_handler.list_run_artifacts():
             return

python_modules/libraries/dagster-dbt/dagster_dbt/cloud_v2/run_handler.py

@@ -98,7 +98,7 @@ def to_default_asset_events(
         manifest: Mapping[str, Any],
         dagster_dbt_translator: Optional[DagsterDbtTranslator] = None,
         context: Optional[AssetExecutionContext] = None,
-    ) -> Iterator[Union[AssetMaterialization, AssetCheckEvaluation, Output]]:
+    ) -> Iterator[Union[AssetCheckEvaluation, AssetCheckResult, AssetMaterialization, Output]]:
         """Convert the run results of a dbt Cloud job run to a set of corresponding Dagster events.
 
         Args:
@@ -109,12 +109,12 @@ def to_default_asset_events(
             context (Optional[AssetExecutionContext]): The execution context.
 
         Returns:
-            Iterator[Union[AssetMaterialization, AssetCheckEvaluation, Output]]:
+            Iterator[Union[AssetCheckEvaluation, AssetCheckResult, AssetMaterialization, Output]]:
                 A set of corresponding Dagster events.
 
                 In a Dagster asset definition, the following are yielded:
                 - Output for refables (e.g. models, seeds, snapshots.)
-                - AssetCheckEvaluation for dbt tests.
+                - AssetCheckResult for dbt tests.
 
                 For ad hoc usage, the following are yielded:
                 - AssetMaterialization for refables (e.g. models, seeds, snapshots.)