fix new ruff rules

flairNLP · Jul 31, 2023 · 3695fe7 · 3695fe7
1 parent a2ab16b
commit 3695fe7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/flair/data.py b/flair/data.py
@@ -143,7 +143,7 @@ def save(self, savefile):
     def __setstate__(self, d):
         self.__dict__ = d
         # set 'add_unk' if the dictionary was created with a version of Flair older than 0.9
-        if "add_unk" not in self.__dict__.keys():
+        if "add_unk" not in self.__dict__:
             self.__dict__["add_unk"] = b"<unk>" in self.__dict__["idx2item"]
 
     @classmethod

diff --git a/flair/embeddings/base.py b/flair/embeddings/base.py
@@ -194,7 +194,7 @@ def embedding_type(self) -> str:
     def _everything_embedded(self, data_points: Sequence[Sentence]) -> bool:
         for sentence in data_points:
             for token in sentence.tokens:
-                if self.name not in token._embeddings.keys():
+                if self.name not in token._embeddings:
                     return False
         return True
 

diff --git a/flair/embeddings/legacy.py b/flair/embeddings/legacy.py
@@ -1330,7 +1330,7 @@ def embed(self, sentences: Union[List[Sentence], Sentence]):
             sentences = [sentences]
 
         for sentence in sentences:
-            if self.name not in sentence._embeddings.keys():
+            if self.name not in sentence._embeddings:
                 everything_embedded = False
 
         if not everything_embedded: