From 25cd1a2b7b37aca06f452aae11d194f70faa17fb Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Tue, 8 Nov 2022 15:35:27 +0800
Subject: [PATCH] Modify module name

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 codebert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codebert.py b/codebert.py
index 2761f97..d31d66e 100644
--- 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.