decomment class

JohnSnowLabs · Apr 8, 2024 · 9467995 · 9467995
1 parent 28ca5a0
commit 9467995
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/nlu/components/embeddings/sentence_mpnet/MPNetSentenceEmbedding.py b/nlu/components/embeddings/sentence_mpnet/MPNetSentenceEmbedding.py
@@ -1,18 +1,18 @@
-# from sparknlp.annotator import MPNetEmbeddings
-#
-#
-# class MPNetSentence:
-#     @staticmethod
-#     def get_default_model():
-#         return MPNetEmbeddings.pretrained() \
-#             .setInputCols(["documents"]) \
-#             .setOutputCol("mpnet_embeddings")
-#
-#     @staticmethod
-#     def get_pretrained_model(name, language, bucket=None):
-#         return MPNetEmbeddings.pretrained(name,language,bucket) \
-#             .setInputCols(["documents"]) \
-#             .setOutputCol("mpnet_embeddings")
-#
-#
-#
+from sparknlp.annotator import MPNetEmbeddings
+
+
+class MPNetSentence:
+    @staticmethod
+    def get_default_model():
+        return MPNetEmbeddings.pretrained() \
+            .setInputCols(["documents"]) \
+            .setOutputCol("mpnet_embeddings")
+
+    @staticmethod
+    def get_pretrained_model(name, language, bucket=None):
+        return MPNetEmbeddings.pretrained(name,language,bucket) \
+            .setInputCols(["documents"]) \
+            .setOutputCol("mpnet_embeddings")
+
+
+