fix

dagster-io · Dec 24, 2024 · ee30a4a · ee30a4a
1 parent 2355dd2
commit ee30a4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...modules/libraries/dagster-components/dagster_components_tests/unit_tests/test_registry.py b/...modules/libraries/dagster-components/dagster_components_tests/unit_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
-    @component(name="test_component_{number}")
+    from dagster_components import Component, component_type
+    @component_type(name="test_component_{number}")
     class TestComponent{number}(Component):
         pass
     """)