add bool fn

dagster-io · Nov 30, 2023 · cfdfd63 · cfdfd63
1 parent 4341d02
commit cfdfd63
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster/_core/execution/context/output.py b/python_modules/dagster/dagster/_core/execution/context/output.py
@@ -242,6 +242,11 @@ def op_def(self) -> "OpDefinition":
 
         return cast(OpDefinition, self._op_def)
 
+    @public
+    @property
+    def has_dagster_type(self) -> bool:
+        return self._dagster_type is not None
+
     @public
     @property
     def dagster_type(self) -> "DagsterType":

diff --git a/python_modules/dagster/dagster/_core/execution/plan/inputs.py b/python_modules/dagster/dagster/_core/execution/plan/inputs.py
@@ -822,7 +822,7 @@ def _load_input_with_input_manager(
 
     if (
         context.upstream_output
-        and context.upstream_output.dagster_type
+        and context.upstream_output.has_dagster_type
         and context.upstream_output.dagster_type.is_nothing
     ):
         yield None