Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions dagster_sqlmesh/resource.py
Original file line number Diff line number Diff line change
Expand Up @@ -369,7 +369,7 @@ def process_events(self, event: console.ConsoleEvent) -> None:

def notify_success(
self, sqlmesh_context: SQLMeshContext
) -> t.Iterator[dg.MaterializeResult]:
) -> t.Iterator[dg.MaterializeResult[t.Any]]:
notify = self._tracker.notify_queue_next()

while notify is not None:
Expand Down Expand Up @@ -412,7 +412,7 @@ def create_materialize_result(
context: dg.AssetExecutionContext,
asset_key: dg.AssetKey,
current_materialization_status: ModelMaterializationStatus,
) -> dg.MaterializeResult:
) -> dg.MaterializeResult[t.Any]:
last_materialization = context.instance.get_latest_materialization_event(
asset_key
)
Expand Down Expand Up @@ -598,7 +598,7 @@ def run(
plan_options: PlanOptions | None = None,
run_options: RunOptions | None = None,
materializations_enabled: bool = True,
) -> t.Iterable[dg.MaterializeResult]:
) -> t.Iterable[dg.MaterializeResult[t.Any]]:
"""Execute SQLMesh based on the configuration given"""
plan_options = plan_options or {}
run_options = run_options or {}
Expand Down
4 changes: 2 additions & 2 deletions sample/dagster_project/definitions.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def get_group_name(self, context, model):


@asset(key=["sources", "reset_asset"])
def reset_asset() -> MaterializeResult:
def reset_asset() -> MaterializeResult[t.Any]:
"""An asset used for testing this entire workflow. If the duckdb database is
found, this will delete it. This allows us to continously test this dag if
this specific asset is materialized
Expand Down Expand Up @@ -108,7 +108,7 @@ def post_full_model() -> pl.DataFrame:
)
def sqlmesh_project(
context: AssetExecutionContext, sqlmesh: SQLMeshResource
) -> t.Iterator[MaterializeResult]:
) -> t.Iterator[MaterializeResult[t.Any]]:
yield from sqlmesh.run(context)


Expand Down
Loading