Have multi_asset_check call builder.create_asset_checks_definition(

dagster-io · Jun 10, 2024 · e8db4ac · e8db4ac
1 parent 4376603
commit e8db4ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/python_modules/dagster/dagster/_core/definitions/decorators/asset_check_decorator.py b/python_modules/dagster/dagster/_core/definitions/decorators/asset_check_decorator.py
@@ -361,14 +361,6 @@ def inner(fn: MultiAssetCheckFunction) -> AssetChecksDefinition:
         )
 
         with disable_dagster_warnings():
-            op_def = builder.create_op_definition()
-
-        return AssetChecksDefinition.create(
-            node_def=op_def,
-            resource_defs=builder.args.op_def_resource_defs,
-            keys_by_input_name=builder.asset_keys_by_input_name,
-            check_specs_by_output_name=builder.check_specs_by_output_name,
-            can_subset=builder.args.can_subset,
-        )
+            return builder.create_asset_checks_definition()
 
     return inner
diff --git a/...dules/dagster/dagster/_core/definitions/decorators/decorator_assets_definition_builder.py b/...dules/dagster/dagster/_core/definitions/decorators/decorator_assets_definition_builder.py
@@ -566,7 +566,7 @@ def create_asset_checks_definition(self) -> AssetChecksDefinition:
             node_def=op_def,
             resource_defs=self.args.assets_def_resource_defs,
             check_specs_by_output_name=self.check_specs_by_output_name,
-            can_subset=False,
+            can_subset=self.args.can_subset,
         )
 
     @cached_property