diff --git a/python_modules/dagster/dagster/_core/definitions/decorators/asset_decorator.py b/python_modules/dagster/dagster/_core/definitions/decorators/asset_decorator.py index a25a8bc1e3d58..1b2f8b36312fe 100644 --- a/python_modules/dagster/dagster/_core/definitions/decorators/asset_decorator.py +++ b/python_modules/dagster/dagster/_core/definitions/decorators/asset_decorator.py @@ -271,12 +271,12 @@ def resolve_asset_key_and_name_for_decorator( key: Optional[CoercibleToAssetKey], key_prefix: Optional[CoercibleToAssetKeyPrefix], name: Optional[str], - decorator: str, + decorator_name: str, fn: Callable[..., Any], ) -> Tuple[AssetKey, str]: if (name or key_prefix) and key: raise DagsterInvalidDefinitionError( - f"Cannot specify a name or key prefix for {decorator} when the key" + f"Cannot specify a name or key prefix for {decorator_name} when the key" " argument is provided." ) key_prefix_list = [key_prefix] if isinstance(key_prefix, str) else key_prefix @@ -377,7 +377,7 @@ def invoke(args: AssetArgs, fn: Callable[..., Any]) -> AssetsDefinition: key_prefix=args.key_prefix, name=args.name, fn=fn, - decorator="@asset", + decorator_name="@asset", ) resource_related_state = ResourceRelatedState( @@ -800,7 +800,7 @@ def graph_asset_no_defaults( key=key, key_prefix=key_prefix, name=name, - decorator="@graph_asset", + decorator_name="@graph_asset", fn=compose_fn, ) diff --git a/python_modules/dagster/dagster/_core/definitions/decorators/assets_definition_factory.py b/python_modules/dagster/dagster/_core/definitions/decorators/assets_definition_factory.py index 2c4becc9e7a8b..b3e25791c9d5a 100644 --- a/python_modules/dagster/dagster/_core/definitions/decorators/assets_definition_factory.py +++ b/python_modules/dagster/dagster/_core/definitions/decorators/assets_definition_factory.py @@ -238,13 +238,13 @@ def __init__( named_outs_by_asset_key: Mapping[AssetKey, NamedOut], internal_deps: Mapping[AssetKey, Set[AssetKey]], op_name: str, - multi_asset_args: AssetsDefinitionBuilderArgs, + args: AssetsDefinitionBuilderArgs, fn: Callable[..., Any], ) -> None: self.named_outs_by_asset_key = named_outs_by_asset_key self.internal_deps = internal_deps self.op_name = op_name - self.args = multi_asset_args + self.args = args self.fn = fn self.named_ins_by_asset_key = ( @@ -339,7 +339,7 @@ def from_specs( named_outs_by_asset_key=named_outs_by_asset_key, internal_deps=internal_deps, op_name=op_name, - multi_asset_args=args, + args=args, fn=fn, ) @@ -403,7 +403,7 @@ def from_asset_outs( named_outs_by_asset_key=output_tuples_by_asset_key, internal_deps=internal_deps, op_name=op_name, - multi_asset_args=args, + args=args, fn=fn, ) diff --git a/python_modules/dagster/dagster/_core/definitions/decorators/source_asset_decorator.py b/python_modules/dagster/dagster/_core/definitions/decorators/source_asset_decorator.py index 3e304b4f8ef74..b952b84b5c178 100644 --- a/python_modules/dagster/dagster/_core/definitions/decorators/source_asset_decorator.py +++ b/python_modules/dagster/dagster/_core/definitions/decorators/source_asset_decorator.py @@ -175,7 +175,7 @@ def __call__(self, observe_fn: SourceAssetObserveFunction) -> SourceAsset: key_prefix=self.key_prefix, name=self.name, fn=observe_fn, - decorator="@observable_source_asset", + decorator_name="@observable_source_asset", ) arg_resource_keys = {arg.name for arg in get_resource_args(observe_fn)}