diff --git a/python_modules/libraries/dagster-components/dagster_components/cli/scaffold.py b/python_modules/libraries/dagster-components/dagster_components/cli/scaffold.py index 1261c7d11cb53..7302bcb132584 100644 --- a/python_modules/libraries/dagster-components/dagster_components/cli/scaffold.py +++ b/python_modules/libraries/dagster-components/dagster_components/cli/scaffold.py @@ -62,9 +62,7 @@ def scaffold_component_command( raise Exception( f"Component type {component_type} does not have a scaffolder. Reason: {scaffolder.message}." ) - scaffold_params = TypeAdapter(scaffolder.get_params_schema_type()).validate_json( - json_params - ) + scaffold_params = TypeAdapter(scaffolder.get_schema()).validate_json(json_params) else: scaffold_params = {} 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 aa48f9ed2e09d..7bef8a61e77aa 100644 --- a/python_modules/libraries/dagster-components/dagster_components/core/component.py +++ b/python_modules/libraries/dagster-components/dagster_components/core/component.py @@ -74,7 +74,7 @@ def get_metadata(cls) -> "ComponentTypeInternalMetadata": ) component_params = cls.get_schema() - scaffold_params = scaffolder.get_params_schema_type() + scaffold_params = scaffolder.get_schema() return { "summary": clean_docstring.split("\n\n")[0] if clean_docstring else None, "description": clean_docstring if clean_docstring else None, diff --git a/python_modules/libraries/dagster-components/dagster_components/core/component_scaffolder.py b/python_modules/libraries/dagster-components/dagster_components/core/component_scaffolder.py index ad336b6ff79c1..0ce5330703710 100644 --- a/python_modules/libraries/dagster-components/dagster_components/core/component_scaffolder.py +++ b/python_modules/libraries/dagster-components/dagster_components/core/component_scaffolder.py @@ -15,7 +15,7 @@ class ComponentScaffoldRequest: class ComponentScaffolder: @classmethod - def get_params_schema_type(cls) -> Optional[type[BaseModel]]: + def get_schema(cls) -> Optional[type[BaseModel]]: return None @abstractmethod diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/scaffolder.py b/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/scaffolder.py index e0e7465411d5c..8efaf5c474d1d 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/scaffolder.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/dbt_project/scaffolder.py @@ -20,7 +20,7 @@ class DbtScaffoldParams(BaseModel): class DbtProjectComponentScaffolder(ComponentScaffolder): @classmethod - def get_params_schema_type(cls) -> Optional[type[BaseModel]]: + def get_schema(cls) -> Optional[type[BaseModel]]: return DbtScaffoldParams def scaffold(self, request: ComponentScaffoldRequest, params: DbtScaffoldParams) -> None: diff --git a/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/scaffolder.py b/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/scaffolder.py index 3b7c6fd7413d8..b723fa884bd37 100644 --- a/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/scaffolder.py +++ b/python_modules/libraries/dagster-components/dagster_components/lib/definitions_component/scaffolder.py @@ -15,7 +15,7 @@ class DefinitionsScaffoldParams(BaseModel): class DefinitionsComponentScaffolder(ComponentScaffolder): @classmethod - def get_params_schema_type(cls): + def get_schema(cls): return DefinitionsScaffoldParams def scaffold( 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 d621f89d9d764..32db23e318318 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 @@ -25,7 +25,7 @@ class SimplePipesScriptAssetParams(BaseModel): class SimplePipesScriptAssetScaffolder(ComponentScaffolder): @classmethod - def get_params_schema_type(cls): + def get_schema(cls): return SimplePipesScriptAssetParams def scaffold(