diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/basic-subclass.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/basic-subclass.py index d2bd46ff10953..61122081d863e 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/basic-subclass.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/basic-subclass.py @@ -1,6 +1,6 @@ -from dagster_components import component_type +from dagster_components import registered_component_type from dagster_components.lib import SlingReplicationCollection -@component_type(name="custom_subclass") +@registered_component_type(name="custom_subclass") class CustomSubclass(SlingReplicationCollection): ... diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/custom-scope.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/custom-scope.py index a8dec75923671..c97a27e04ac68 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/custom-scope.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/custom-scope.py @@ -1,13 +1,13 @@ from collections.abc import Mapping from typing import Any -from dagster_components import component_type +from dagster_components import registered_component_type from dagster_components.lib import SlingReplicationCollection import dagster as dg -@component_type(name="custom_subclass") +@registered_component_type(name="custom_subclass") class SubclassWithScope(SlingReplicationCollection): def get_additional_scope(self) -> Mapping[str, Any]: def _custom_cron(cron_schedule: str) -> dg.AutomationCondition: diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/debug-mode.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/debug-mode.py index bca987009b048..2e13f4341650b 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/debug-mode.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/custom-subclass/debug-mode.py @@ -1,13 +1,13 @@ from collections.abc import Iterator -from dagster_components import component_type +from dagster_components import registered_component_type from dagster_components.lib import SlingReplicationCollection from dagster_sling import SlingResource import dagster as dg -@component_type(name="debug_sling_replication") +@registered_component_type(name="debug_sling_replication") class DebugSlingReplicationComponent(SlingReplicationCollection): def execute( self, context: dg.AssetExecutionContext, sling: SlingResource diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/empty.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/empty.py index 568db62e85b32..46cf43adc20d2 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/empty.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/empty.py @@ -1,10 +1,14 @@ -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import ( + Component, + ComponentLoadContext, + registered_component_type, +) from pydantic import BaseModel from dagster import Definitions -@component_type(name="shell_command") +@registered_component_type(name="shell_command") class ShellCommand(Component): @classmethod def get_schema(cls) -> type[BaseModel]: ... diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/resolving-resolvable-field.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/resolving-resolvable-field.py index e2dee1a2ba6c8..6f311cc0659ef 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/resolving-resolvable-field.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/resolving-resolvable-field.py @@ -1,4 +1,8 @@ -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import ( + Component, + ComponentLoadContext, + registered_component_type, +) import dagster as dg @@ -10,7 +14,7 @@ def _get_script_runner(val: str) -> ScriptRunner: return ScriptRunner() -@component_type(name="shell_command") +@registered_component_type(name="shell_command") class ShellCommand(Component): def __init__(self, params): self.params = params diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-build-defs.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-build-defs.py index ffd17272a8282..aac20ba0d01b7 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-build-defs.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-build-defs.py @@ -1,7 +1,11 @@ import subprocess from typing import Optional -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import ( + Component, + ComponentLoadContext, + registered_component_type, +) from dagster_components.core.schema.objects import AssetAttributesModel, OpSpecModel from pydantic import BaseModel @@ -16,7 +20,7 @@ class ShellScriptSchema(BaseModel): # highlight-end -@component_type(name="shell_command") +@registered_component_type(name="shell_command") class ShellCommand(Component): def __init__(self, params: ShellScriptSchema): self.params = params diff --git a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-config-schema.py b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-config-schema.py index 3e7bbccfadbbe..2be5c1d0bdf14 100644 --- a/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-config-schema.py +++ b/examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/with-config-schema.py @@ -1,6 +1,10 @@ from typing import Optional -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import ( + Component, + ComponentLoadContext, + registered_component_type, +) from dagster_components.core.schema.objects import AssetAttributesModel, OpSpecModel from pydantic import BaseModel @@ -15,7 +19,7 @@ class ShellScriptSchema(BaseModel): # highlight-end -@component_type(name="shell_command") +@registered_component_type(name="shell_command") class ShellCommand(Component): @classmethod def get_schema(cls) -> type[ShellScriptSchema]: diff --git a/python_modules/libraries/dagster-components/dagster_components/__init__.py b/python_modules/libraries/dagster-components/dagster_components/__init__.py index e43da96201f74..fe4c36d9cb363 100644 --- a/python_modules/libraries/dagster-components/dagster_components/__init__.py +++ b/python_modules/libraries/dagster-components/dagster_components/__init__.py @@ -2,7 +2,7 @@ Component as Component, ComponentLoadContext as ComponentLoadContext, ComponentTypeRegistry as ComponentTypeRegistry, - component_type as component_type, + registered_component_type as registered_component_type, ) from dagster_components.core.component_defs_builder import ( build_component_defs as build_component_defs, diff --git a/python_modules/libraries/dagster-components/dagster_components/core/component.py b/python_modules/libraries/dagster-components/dagster_components/core/component.py index df9583fe98a26..9915ee97a7b5a 100644 --- a/python_modules/libraries/dagster-components/dagster_components/core/component.py +++ b/python_modules/libraries/dagster-components/dagster_components/core/component.py @@ -254,7 +254,9 @@ def for_decl_node(self, decl_node: ComponentDeclNode) -> "ComponentLoadContext": COMPONENT_LOADER_FN_ATTR = "__dagster_component_loader_fn" -def component_type(cls: Optional[type[Component]] = None, *, name: Optional[str] = None) -> Any: +def registered_component_type( + cls: Optional[type[Component]] = None, *, name: Optional[str] = None +) -> Any: """Decorator for registering a component type. You must annotate a component type with this decorator in order for it to be inspectable and loaded by tools. diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/component.py b/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/component.py index 43dfc630059ef..6fe4cc4303e6a 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/component.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/component.py @@ -12,7 +12,7 @@ ) from dagster_components import Component, ComponentLoadContext -from dagster_components.core.component import component_type +from dagster_components.core.component import registered_component_type from dagster_components.core.schema.base import ResolvableModel from dagster_components.core.schema.metadata import ResolvableFieldInfo from dagster_components.core.schema.objects import ( @@ -46,7 +46,7 @@ def resolve(self, resolver: TemplatedValueResolver) -> "DbtProjectComponent": ) -@component_type(name="dbt_project") +@registered_component_type(name="dbt_project") class DbtProjectComponent(Component): """Expose a DBT project to Dagster as a set of assets.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/component.py b/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/component.py index 6484e813003e4..eb5c5e850750a 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/component.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/component.py @@ -10,7 +10,7 @@ from pydantic import BaseModel from typing_extensions import Self -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import Component, ComponentLoadContext, registered_component_type from dagster_components.lib.definitions_component.scaffolder import DefinitionsComponentScaffolder @@ -18,7 +18,7 @@ class DefinitionsParamSchema(BaseModel): definitions_path: Optional[str] = None -@component_type(name="definitions") +@registered_component_type(name="definitions") class DefinitionsComponent(Component): """Wraps an arbitrary set of Dagster definitions.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/pipes_subprocess_script_collection.py b/python_modules/libraries/dagster-components/dagster_components/lib/pipes_subprocess_script_collection.py index 2a6d026fe488b..c04915d6a4353 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/pipes_subprocess_script_collection.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/pipes_subprocess_script_collection.py @@ -10,7 +10,11 @@ from dagster._core.pipes.subprocess import PipesSubprocessClient from pydantic import BaseModel -from dagster_components.core.component import Component, ComponentLoadContext, component_type +from dagster_components.core.component import ( + Component, + ComponentLoadContext, + registered_component_type, +) from dagster_components.core.schema.base import ResolvableModel from dagster_components.core.schema.objects import AssetSpecModel from dagster_components.core.schema.resolver import TemplatedValueResolver @@ -36,7 +40,7 @@ def resolve(self, resolver: TemplatedValueResolver) -> Mapping[str, Sequence[Ass } -@component_type(name="pipes_subprocess_script_collection") +@registered_component_type(name="pipes_subprocess_script_collection") class PipesSubprocessScriptCollection(Component): """Assets that wrap Python scripts executed with Dagster's PipesSubprocessClient.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/sling_replication_collection/component.py b/python_modules/libraries/dagster-components/dagster_components/lib/sling_replication_collection/component.py index 4245c6f22c383..9715d8bd31772 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/sling_replication_collection/component.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/sling_replication_collection/component.py @@ -12,7 +12,7 @@ from typing_extensions import Self from dagster_components import Component, ComponentLoadContext -from dagster_components.core.component import component_type +from dagster_components.core.component import registered_component_type from dagster_components.core.component_scaffolder import ComponentScaffolder from dagster_components.core.schema.metadata import ResolvableFieldInfo from dagster_components.core.schema.objects import ( @@ -73,7 +73,7 @@ def resolve(self, resolver: TemplatedValueResolver) -> ResolvedSlingReplicationC ) -@component_type +@registered_component_type class SlingReplicationCollection(Component): """Expose one or more Sling replications to Dagster as assets.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/test/all_metadata_empty_asset.py b/python_modules/libraries/dagster-components/dagster_components/lib/test/all_metadata_empty_asset.py index 140d636020f11..73e9436faf97d 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/test/all_metadata_empty_asset.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/test/all_metadata_empty_asset.py @@ -3,11 +3,11 @@ from dagster._core.execution.context.asset_execution_context import AssetExecutionContext from typing_extensions import Self -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import Component, ComponentLoadContext, registered_component_type from dagster_components.core.component_scaffolder import DefaultComponentScaffolder -@component_type(name="all_metadata_empty_asset") +@registered_component_type(name="all_metadata_empty_asset") class AllMetadataEmptyAsset(Component): @classmethod def load(cls, context: "ComponentLoadContext") -> Self: diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/test/complex_schema_asset.py b/python_modules/libraries/dagster-components/dagster_components/lib/test/complex_schema_asset.py index 43788b6e7994e..d1cb2404e20a4 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/test/complex_schema_asset.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/test/complex_schema_asset.py @@ -7,7 +7,7 @@ from pydantic import BaseModel from typing_extensions import Self -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import Component, ComponentLoadContext, registered_component_type from dagster_components.core.component_scaffolder import DefaultComponentScaffolder from dagster_components.core.schema.metadata import ResolvableFieldInfo from dagster_components.core.schema.objects import ( @@ -26,7 +26,7 @@ class ComplexAssetParams(BaseModel): asset_transforms: Optional[Sequence[AssetSpecTransformModel]] = None -@component_type(name="complex_schema_asset") +@registered_component_type(name="complex_schema_asset") class ComplexSchemaAsset(Component): """An asset that has a complex params schema.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_asset.py b/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_asset.py index 04dabf9815415..51edffade05ed 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_asset.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_asset.py @@ -5,7 +5,7 @@ from pydantic import BaseModel from typing_extensions import Self -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import Component, ComponentLoadContext, registered_component_type from dagster_components.core.component_scaffolder import ( ComponentScaffolder, DefaultComponentScaffolder, @@ -17,7 +17,7 @@ class SimpleAssetParams(BaseModel): value: str -@component_type(name="simple_asset") +@registered_component_type(name="simple_asset") class SimpleAsset(Component): """A simple asset that returns a constant string value.""" diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_pipes_script_asset.py b/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_pipes_script_asset.py index 41d9f0d88bb9b..4a3e91e6348ec 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_pipes_script_asset.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/test/simple_pipes_script_asset.py @@ -9,7 +9,7 @@ from pydantic import BaseModel from typing_extensions import Self -from dagster_components import Component, ComponentLoadContext, component_type +from dagster_components import Component, ComponentLoadContext, registered_component_type from dagster_components.core.component_scaffolder import ( ComponentScaffolder, ComponentScaffoldRequest, @@ -47,7 +47,7 @@ def scaffold( """ -@component_type(name="simple_pipes_script_asset") +@registered_component_type(name="simple_pipes_script_asset") class SimplePipesScriptAsset(Component): """A simple asset that runs a Python script with the Pipes subprocess client. diff --git a/python_modules/libraries/dagster-components/dagster_components/test/basic_components.py b/python_modules/libraries/dagster-components/dagster_components/test/basic_components.py index e79447e29e8de..2a291e10979f0 100644 --- a/python_modules/libraries/dagster-components/dagster_components/test/basic_components.py +++ b/python_modules/libraries/dagster-components/dagster_components/test/basic_components.py @@ -6,7 +6,7 @@ from pydantic import BaseModel, ConfigDict from typing_extensions import Self -from dagster_components import Component, component_type +from dagster_components import Component, registered_component_type from dagster_components.core.component import ComponentLoadContext @@ -17,7 +17,7 @@ class MyComponentSchema(BaseModel): model_config = ConfigDict(extra="forbid") -@component_type +@registered_component_type class MyComponent(Component): name = "my_component" @@ -46,7 +46,7 @@ class MyNestedComponentSchema(BaseModel): model_config = ConfigDict(extra="forbid") -@component_type +@registered_component_type class MyNestedComponent(Component): name = "my_nested_component" diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/code_locations/custom_sling_location/custom_sling_location/components/debug_sling_component/component.py b/python_modules/libraries/dagster-components/dagster_components_tests/code_locations/custom_sling_location/custom_sling_location/components/debug_sling_component/component.py index a22253924e935..0da084a8fec22 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/code_locations/custom_sling_location/custom_sling_location/components/debug_sling_component/component.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/code_locations/custom_sling_location/custom_sling_location/components/debug_sling_component/component.py @@ -1,12 +1,12 @@ from collections.abc import Iterator from dagster._core.execution.context.asset_execution_context import AssetExecutionContext -from dagster_components import component_type +from dagster_components import registered_component_type from dagster_components.lib.sling_replication_collection.component import SlingReplicationCollection from dagster_sling import SlingResource -@component_type(name="debug_sling_replication") +@registered_component_type(name="debug_sling_replication") class DebugSlingReplicationComponent(SlingReplicationCollection): def execute(self, context: AssetExecutionContext, sling: SlingResource) -> Iterator: return sling.replicate(context=context, debug=True) diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/local_component_sample/__init__.py b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/local_component_sample/__init__.py index fb45b00e6814a..6011586f3cff3 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/local_component_sample/__init__.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/local_component_sample/__init__.py @@ -1,5 +1,5 @@ from dagster._core.definitions.definitions_class import Definitions -from dagster_components import Component, component_type +from dagster_components import Component, registered_component_type from dagster_components.core.component import ComponentLoadContext from dagster_components.core.schema.base import BaseModel from typing_extensions import Self @@ -10,7 +10,7 @@ class MyComponentSchema(BaseModel): an_int: int -@component_type +@registered_component_type class MyComponent(Component): name = "my_component" diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/other_local_component_sample/__init__.py b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/other_local_component_sample/__init__.py index 42d81a6d640f8..c29b1bba17f81 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/other_local_component_sample/__init__.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/components/definitions/other_local_component_sample/__init__.py @@ -1,5 +1,5 @@ from dagster._core.definitions.definitions_class import Definitions -from dagster_components import Component, component_type +from dagster_components import Component, registered_component_type from dagster_components.core.component import ComponentLoadContext from pydantic import BaseModel from typing_extensions import Self @@ -10,7 +10,7 @@ class MyNewComponentSchema(BaseModel): an_int: int -@component_type +@registered_component_type class MyNewComponent(Component): name = "my_new_component" diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/test_sling_integration_test.py b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/test_sling_integration_test.py index aad4243aea609..8fc1aae6279ad 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/test_sling_integration_test.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/integration_tests/test_sling_integration_test.py @@ -12,7 +12,7 @@ from dagster._core.definitions.result import MaterializeResult from dagster._core.execution.context.asset_execution_context import AssetExecutionContext from dagster._utils.env import environ -from dagster_components import component_type +from dagster_components import registered_component_type from dagster_components.core.component_decl_builder import ComponentFileModel from dagster_components.core.component_defs_builder import ( YamlComponentDecl, @@ -163,7 +163,7 @@ def test_load_from_path(sling_path: Path) -> None: def test_sling_subclass() -> None: - @component_type(name="debug_sling_replication") + @registered_component_type(name="debug_sling_replication") class DebugSlingReplicationComponent(SlingReplicationCollection): def execute( self, context: AssetExecutionContext, sling: SlingResource diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/registry_tests/test_registry.py b/python_modules/libraries/dagster-components/dagster_components_tests/registry_tests/test_registry.py index f237d977f813a..9c20d3b3e9a86 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/registry_tests/test_registry.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/registry_tests/test_registry.py @@ -55,8 +55,8 @@ def _find_repo_root(): def _generate_test_component_source(number: int) -> str: return textwrap.dedent(f""" - from dagster_components import Component, component_type - @component_type(name="test_component_{number}") + from dagster_components import Component, registered_component_type + @registered_component_type(name="test_component_{number}") class TestComponent{number}(Component): pass """) diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/rendering_tests/custom_scope_component/component.py b/python_modules/libraries/dagster-components/dagster_components_tests/rendering_tests/custom_scope_component/component.py index 72391a47321a7..aef3ff5eaf6d1 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/rendering_tests/custom_scope_component/component.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/rendering_tests/custom_scope_component/component.py @@ -2,7 +2,12 @@ from typing import Any from dagster import AssetSpec, AutomationCondition, Definitions -from dagster_components import AssetAttributesModel, Component, ComponentLoadContext, component_type +from dagster_components import ( + AssetAttributesModel, + Component, + ComponentLoadContext, + registered_component_type, +) def my_custom_fn(a: str, b: str) -> str: @@ -13,7 +18,7 @@ def my_custom_automation_condition(cron_schedule: str) -> AutomationCondition: return AutomationCondition.cron_tick_passed(cron_schedule) & ~AutomationCondition.in_progress() -@component_type(name="custom_scope_component") +@registered_component_type(name="custom_scope_component") class HasCustomScope(Component): @classmethod def get_additional_scope(cls) -> Mapping[str, Any]: diff --git a/python_modules/libraries/dagster-components/dagster_components_tests/unit_tests/test_registered_component.py b/python_modules/libraries/dagster-components/dagster_components_tests/unit_tests/test_registered_component.py index f53b7dd12038a..bcef8659a3661 100644 --- a/python_modules/libraries/dagster-components/dagster_components_tests/unit_tests/test_registered_component.py +++ b/python_modules/libraries/dagster-components/dagster_components_tests/unit_tests/test_registered_component.py @@ -1,9 +1,9 @@ -from dagster_components import Component, component_type +from dagster_components import Component, registered_component_type from dagster_components.core.component import get_component_type_name, is_registered_component_type def test_registered_component_with_default_name() -> None: - @component_type + @registered_component_type class RegisteredComponent(Component): ... assert is_registered_component_type(RegisteredComponent) @@ -11,7 +11,7 @@ class RegisteredComponent(Component): ... def test_registered_component_with_default_name_and_parens() -> None: - @component_type() + @registered_component_type() class RegisteredComponent(Component): ... assert is_registered_component_type(RegisteredComponent) @@ -19,7 +19,7 @@ class RegisteredComponent(Component): ... def test_registered_component_with_explicit_kwarg_name() -> None: - @component_type(name="explicit_name") + @registered_component_type(name="explicit_name") class RegisteredComponent(Component): ... assert is_registered_component_type(RegisteredComponent) diff --git a/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja b/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja index 8933a53af027e..7a2a1798defc6 100644 --- a/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja +++ b/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja @@ -3,14 +3,14 @@ from dagster_components import ( Component, ComponentLoadContext, DefaultComponentScaffolder, - component_type, + registered_component_type, ) from pydantic import BaseModel class {{ component_type_class_name }}Params(BaseModel): ... -@component_type(name="{{ name }}") +@registered_component_type(name="{{ name }}") class {{ component_type_class_name }}(Component): """COMPONENT SUMMARY HERE.