deepdata/runtime/ops/examples/test_operator/process.py

from typing import Dict, Any

from deepdata.core.base_op import Mapper


class TestMapper(Mapper):
    """
    算�类�建议使用驼峰命�法定义，例如 TestMapper
    """

    def __init__(self, *args, **kwargs):
        """
        �始化��        :param args:
        :param kwargs:
        """
        super().__init__(*args, **kwargs)
        self.slider_param = float(kwargs.get("sliderParam", 0.5))
        self.switch_param = kwargs.get('switchParam', False)
        self.select_param = kwargs.get('selectParam', '')
        self.radio_param = kwargs.get('radioParam', '')
        self.range_param = kwargs.get('rangeParam', [0, 0])
        self.checkbox_param = kwargs.get('checkboxParam', [])
        self.input_param = kwargs.get('inputParam', '').strip()

    def execute(self, sample: Dict[str, Any]) -> Dict[str, Any]:
        """
        核心处�逻辑
        :param sample: 输入的数�样本，通常包� text_key 等字�        :return: 处��的数�样本
        """
        # 示例：获�文本并进行修改
        # input_text = sample['text']
        # processed_text = do_something(input_text)
        # sample['text'] = processed_text

        return sample