mirror of
https://github.com/NVIDIA-NeMo/DataDesigner
synced 2026-05-24 09:48:29 +00:00
f8c201e085
* update script * update headers * refactor a bit and add test script * update headers * update for edge case * update headers * add step to get file creation date * use git history to get copyright year * generation type is printed with inference parameters * fix unit test
18 lines
863 B
Python
18 lines
863 B
Python
# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from data_designer.config.processors import DropColumnsProcessorConfig, ProcessorType
|
|
from data_designer.engine.processing.processors.drop_columns import DropColumnsProcessor
|
|
from data_designer.engine.processing.processors.registry import (
|
|
ProcessorRegistry,
|
|
create_default_processor_registry,
|
|
)
|
|
|
|
|
|
def test_create_default_processor_registry():
|
|
registry = create_default_processor_registry()
|
|
|
|
assert isinstance(registry, ProcessorRegistry)
|
|
assert ProcessorType.DROP_COLUMNS in ProcessorRegistry._registry
|
|
assert ProcessorRegistry._registry[ProcessorType.DROP_COLUMNS] == DropColumnsProcessor
|
|
assert ProcessorRegistry._config_registry[ProcessorType.DROP_COLUMNS] == DropColumnsProcessorConfig
|