format issue

QData · Mar 1, 2022 · 92374d3 · 92374d3
1 parent 613362e
commit 92374d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/textattack/constraints/semantics/sentence_encoders/thought_vector.py b/textattack/constraints/semantics/sentence_encoders/thought_vector.py
@@ -19,9 +19,7 @@ class ThoughtVector(SentenceEncoder):
         word_embedding (textattack.shared.AbstractWordEmbedding): The word embedding to use
     """
 
-    def __init__(
-        self, embedding=None, **kwargs
-    ):
+    def __init__(self, embedding=None, **kwargs):
         if embedding is None:
             embedding = WordEmbedding.counterfitted_GLOVE_embedding()
         if not isinstance(embedding, AbstractWordEmbedding):

diff --git a/textattack/transformations/word_swaps/word_swap_embedding.py b/textattack/transformations/word_swaps/word_swap_embedding.py
@@ -28,12 +28,7 @@ class WordSwapEmbedding(WordSwap):
     >>> augmenter.augment(s)
     """
 
-    def __init__(
-        self,
-        max_candidates=15,
-        embedding=None,
-        **kwargs
-    ):
+    def __init__(self, max_candidates=15, embedding=None, **kwargs):
         super().__init__(**kwargs)
         if embedding is None:
             embedding = WordEmbedding.counterfitted_GLOVE_embedding()