black formatting and ruff fixes

flairNLP · Jul 17, 2023 · 7f79e92 · 7f79e92
1 parent e93a75f
commit 7f79e92
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/flair/data.py b/flair/data.py
@@ -967,7 +967,7 @@ def to_dict(self, tag_type: Optional[str] = None):
             "labels": [label.to_dict() for label in self.get_labels(tag_type) if label.data_point is self],
             "entities": [span.to_dict(tag_type) for span in self.get_spans(tag_type)],
             "relations": [relation.to_dict(tag_type) for relation in self.get_relations(tag_type)],
-            "tokens": [token.to_dict(tag_type) for token in self.tokens]
+            "tokens": [token.to_dict(tag_type) for token in self.tokens],
         }
 
     def get_span(self, start: int, stop: int):

diff --git a/flair/training_utils.py b/flair/training_utils.py
@@ -66,12 +66,7 @@ def micro_avg_f_score(self):
         return self.mean_squared_error()
 
     def to_tsv(self):
-        return "{}\t{}\t{}\t{}".format(
-            self.mean_squared_error(),
-            self.mean_absolute_error(),
-            self.pearsonr(),
-            self.spearmanr(),
-        )
+        return f"{self.mean_squared_error()}\t{self.mean_absolute_error()}\t{self.pearsonr()}\t{self.spearmanr()}"
 
     @staticmethod
     def tsv_header(prefix=None):

diff --git a/pyproject.toml b/pyproject.toml
@@ -85,7 +85,7 @@ ignore = [
   "D105",
   "D107",
   "E501",   # Ignore line too long
-  "RUFF012",
+  "RUF012",
 ]
 
 unfixable = [