Modify module name

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 25cd1a2b7b
1 changed files with 1 additions and 1 deletions
--- a/codebert.py
+++ b/codebert.py
@ -30,7 +30,7 @@ log = logging.getLogger()


@register(output_schema=['vec'])
-class AutoTransformers(NNOperator):
+class CodeBert(NNOperator):
    """
    An operator generates an embedding for code or natural language text
    using a pretrained codebert model gathered by huggingface.