decouple delete class from config

Author: sbilge

src/metldata/builtin_transformations/delete_class/assumptions.py

@@ -20,21 +20,15 @@
 from metldata.builtin_transformations.common.assumptions.path_assumptions import (
     check_class_exists,
 )
-from metldata.builtin_transformations.delete_class.config import (
-    DeleteClassConfig,
-)
 from metldata.transform.exceptions import ModelAssumptionError
 
 
-def check_model_assumptions(
-    model: SchemaPack,
-    transformation_config: DeleteClassConfig,
-) -> None:
+def check_model_assumptions(model: SchemaPack, class_name: str) -> None:
     """Check model assumptions for the delete class transformation."""
-    check_class_exists(model=model, class_name=transformation_config.class_name)
+    check_class_exists(model=model, class_name=class_name)
     check_class_is_not_root(
         model=model,
-        class_name=transformation_config.class_name,
+        class_name=class_name,
     )
 
 

src/metldata/builtin_transformations/delete_class/data_transform.py

@@ -18,27 +18,22 @@
 from schemapack.spec.datapack import DataPack
 
 from metldata.builtin_transformations.common.utils import data_to_dict
-from metldata.builtin_transformations.delete_class.config import (
-    DeleteClassConfig,
-)
 from metldata.transform.exceptions import EvitableTransformationError
 
 
-def delete_data_class(
-    *, data: DataPack, transformation_config: DeleteClassConfig
-) -> DataPack:
+def delete_data_class(*, data: DataPack, class_name: str) -> DataPack:
     """Delete class from the provided data."""
     modified_data = data_to_dict(data)
 
     try:
-        del modified_data["resources"][transformation_config.class_name]
+        del modified_data["resources"][class_name]
     except KeyError as exc:
         raise EvitableTransformationError() from exc
 
     _remove_relations_from_data(
         modified_data=modified_data,
         original_data=data,
-        target_class=transformation_config.class_name,
+        target_class=class_name,
     )
     return DataPack.model_validate(modified_data)
 

src/metldata/builtin_transformations/delete_class/main.py

@@ -50,7 +50,7 @@ def transform(self, data: DataPack, annotation: SubmissionAnnotation) -> DataPac
             DataTransformationError:
                 if the transformation fails.
         """
-        return delete_data_class(data=data, transformation_config=self._config)
+        return delete_data_class(data=data, class_name=self._config.class_name)
 
 
 def check_model_assumptions_wrapper(
@@ -62,7 +62,7 @@ def check_model_assumptions_wrapper(
         ModelAssumptionError:
             if the model does not fulfill the assumptions.
     """
-    check_model_assumptions(model=model, transformation_config=config)
+    check_model_assumptions(model=model, class_name=config.class_name)
 
 
 def transform_model(model: SchemaPack, config: DeleteClassConfig) -> SchemaPack:
@@ -72,7 +72,7 @@ def transform_model(model: SchemaPack, config: DeleteClassConfig) -> SchemaPack:
         DataModelTransformationError:
             if the transformation fails.
     """
-    return delete_model_class(model=model, transformation_config=config)
+    return delete_model_class(model=model, class_name=config.class_name)
 
 
 DELETE_CLASS_TRANSFORMATION = TransformationDefinition[DeleteClassConfig](

src/metldata/builtin_transformations/delete_class/model_transform.py

@@ -18,29 +18,22 @@
 from schemapack.spec.schemapack import SchemaPack
 
 from metldata.builtin_transformations.common.utils import model_to_dict
-from metldata.builtin_transformations.delete_class.config import (
-    DeleteClassConfig,
-)
 from metldata.transform.exceptions import EvitableTransformationError
 
 
-def delete_model_class(
-    *,
-    model: SchemaPack,
-    transformation_config: DeleteClassConfig,
-) -> SchemaPack:
+def delete_model_class(*, model: SchemaPack, class_name: str) -> SchemaPack:
     """Delete class from the model."""
     mutable_model = model_to_dict(model)
 
     try:
-        del mutable_model["classes"][transformation_config.class_name]
+        del mutable_model["classes"][class_name]
     except KeyError as exc:
         raise EvitableTransformationError() from exc
 
     _remove_relations_from_model(
         mutable_model=mutable_model,
         original_model=model,
-        target_class=transformation_config.class_name,
+        target_class=class_name,
     )
 
     return SchemaPack.model_validate(mutable_model)