[dagster-contrib-modal] support AssetExecutionContext in pipes run

dagster-io · Dec 11, 2024 · 531e555 · 531e555
1 parent 1106a5a
commit 531e555
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/libraries/dagster-contrib-modal/dagster_contrib_modal/resources.py b/libraries/dagster-contrib-modal/dagster_contrib_modal/resources.py
@@ -3,7 +3,7 @@
 from pathlib import Path
 from typing import Mapping, Optional, Union
 
-from dagster import OpExecutionContext, PipesSubprocessClient
+from dagster import AssetExecutionContext, OpExecutionContext, PipesSubprocessClient
 from dagster._annotations import public
 from dagster._core.pipes.client import PipesClientCompletedInvocation
 from dagster_pipes import PipesExtras
@@ -46,7 +46,7 @@ def run(
         self,
         *,
         func_ref: str,
-        context: OpExecutionContext,
+        context: Union[AssetExecutionContext, OpExecutionContext],
         extras: Optional[PipesExtras] = None,
         env: Optional[Mapping[str, str]] = None,
     ) -> PipesClientCompletedInvocation:

diff --git a/libraries/dagster-contrib-modal/pyproject.toml b/libraries/dagster-contrib-modal/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "dagster-contrib-modal"
-version = "0.0.1"
+version = "0.0.2"
 description = "Dagster integration with Modal"
 readme = "README.md"
 requires-python = ">=3.8"

diff --git a/libraries/dagster-contrib-modal/uv.lock b/libraries/dagster-contrib-modal/uv.lock