新算子
175 字小于 1 分钟
2025-06-12
新算子
DataFlow算子有统一基类OperatorABC
,位于dataflow/operators/generate
,dataflow/operators/filter
或dataflow/operators/refine
下,这些算子需要实现基类所需的固定方法run()
如果想要在DataFlow中添加新算子,在实现算子及其所包含的方法后需要进行如下操作:
在算子所在文件夹下添加包含算子类的文件
在该文件中导入Registry实例,并用
register()
方法修饰在算子所在文件夹下的__init__.py文件中,向
_import_structure
变量添加算子指向的相对路径。
如果有必要添加新的算子文件夹,需要在dataflow/utils/registry.py
中进行相应修改。