Merge pull request #3295 from flairNLP/rename-entity-linker_2

Rename EntityLinker and remove some legacy embeddings
flairNLP · Aug 8, 2023 · 88a23be · 88a23be
2 parents b18aff2 + 0fb6cb2
commit 88a23be
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 947 deletions.
diff --git a/flair/embeddings/__init__.py b/flair/embeddings/__init__.py
@@ -32,19 +32,10 @@
 
 # Expose legacy embedding classes
 from .legacy import (
-    BertEmbeddings,
-    CamembertEmbeddings,
     CharLMEmbeddings,
     DocumentLSTMEmbeddings,
     DocumentMeanEmbeddings,
     ELMoEmbeddings,
-    ELMoTransformerEmbeddings,
-    OpenAIGPT2Embeddings,
-    OpenAIGPTEmbeddings,
-    RoBERTaEmbeddings,
-    XLMEmbeddings,
-    XLMRobertaEmbeddings,
-    XLNetEmbeddings,
 )
 
 # Expose token embedding classes
@@ -81,18 +72,9 @@
     "ImageEmbeddings",
     "NetworkImageEmbeddings",
     "PrecomputedImageEmbeddings",
-    "BertEmbeddings",
-    "CamembertEmbeddings",
     "CharLMEmbeddings",
     "DocumentLSTMEmbeddings",
     "DocumentMeanEmbeddings",
-    "ELMoTransformerEmbeddings",
-    "OpenAIGPT2Embeddings",
-    "OpenAIGPTEmbeddings",
-    "RoBERTaEmbeddings",
-    "XLMEmbeddings",
-    "XLMRobertaEmbeddings",
-    "XLNetEmbeddings",
     "BPEmbSerializable",
     "BytePairEmbeddings",
     "CharacterEmbeddings",