lint

Author: andreatgretel

src/data_designer/config/exports.py

@@ -33,9 +33,9 @@
     UniformDistributionParams,
 )
 from data_designer.config.processors import (
-    SchemaTransformProcessorConfig,
     DropColumnsProcessorConfig,
     ProcessorType,
+    SchemaTransformProcessorConfig,
 )
 from data_designer.config.sampler_constraints import ColumnInequalityConstraint, ScalarInequalityConstraint
 from data_designer.config.sampler_params import (

src/data_designer/engine/processing/processors/registry.py

@@ -3,13 +3,13 @@
 
 from data_designer.config.base import ConfigBase
 from data_designer.config.processors import (
-    SchemaTransformProcessorConfig,
     DropColumnsProcessorConfig,
     ProcessorType,
+    SchemaTransformProcessorConfig,
 )
-from data_designer.engine.processing.processors.schema_transform import SchemaTransformProcessor
 from data_designer.engine.processing.processors.base import Processor
 from data_designer.engine.processing.processors.drop_columns import DropColumnsProcessor
+from data_designer.engine.processing.processors.schema_transform import SchemaTransformProcessor
 from data_designer.engine.registry.base import TaskRegistry
 
 
@@ -18,8 +18,6 @@ class ProcessorRegistry(TaskRegistry[str, Processor, ConfigBase]): ...
 
 def create_default_processor_registry() -> ProcessorRegistry:
     registry = ProcessorRegistry()
-    registry.register(
-        ProcessorType.SCHEMA_TRANSFORM, SchemaTransformProcessor, SchemaTransformProcessorConfig, False
-    )
+    registry.register(ProcessorType.SCHEMA_TRANSFORM, SchemaTransformProcessor, SchemaTransformProcessorConfig, False)
     registry.register(ProcessorType.DROP_COLUMNS, DropColumnsProcessor, DropColumnsProcessorConfig, False)
     return registry

tests/config/test_processors.py

@@ -6,10 +6,10 @@
 
 from data_designer.config.dataset_builders import BuildStage
 from data_designer.config.processors import (
-    SchemaTransformProcessorConfig,
     DropColumnsProcessorConfig,
     ProcessorConfig,
     ProcessorType,
+    SchemaTransformProcessorConfig,
     get_processor_config_from_kwargs,
 )
 

tests/config/utils/test_validation.py

@@ -15,21 +15,21 @@
 from data_designer.config.dataset_builders import BuildStage
 from data_designer.config.models import ImageContext, ModalityDataType
 from data_designer.config.processors import (
-    SchemaTransformProcessorConfig,
     DropColumnsProcessorConfig,
+    SchemaTransformProcessorConfig,
 )
 from data_designer.config.utils.code_lang import CodeLang
 from data_designer.config.utils.validation import (
     Violation,
     ViolationLevel,
     ViolationType,
     rich_print_violations,
-    validate_schema_transform_processor,
     validate_code_validation,
     validate_columns_not_all_dropped,
     validate_data_designer_config,
     validate_expression_references,
     validate_prompt_templates,
+    validate_schema_transform_processor,
 )
 from data_designer.config.validator_params import CodeValidatorParams