fix format issue in chinse_word_swap_masked.py

QData · Mar 11, 2024 · df5f14d · df5f14d
1 parent e8f74bc
commit df5f14d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -20,4 +20,3 @@ more-itertools
 pinyin>=0.4.0
 jieba
 OpenHowNet
-click==8.0.2
diff --git a/textattack/transformations/word_swaps/chn_transformations/chinese_word_swap_masked.py b/textattack/transformations/word_swaps/chn_transformations/chinese_word_swap_masked.py
@@ -3,7 +3,7 @@
 -------------------------------------
 """
 
-from transformers import pipeline
+import torch
 
 from . import WordSwap
 
@@ -13,8 +13,7 @@ class ChineseWordSwapMaskedLM(WordSwap):
     model."""
 
     def __init__(self, task="fill-mask", model="xlm-roberta-base", **kwargs):
-        from transformers import BertTokenizer, BertForMaskedLM
-        import torch
+        from transformers import BertForMaskedLM, BertTokenizer
 
         self.tt = BertTokenizer.from_pretrained(model)
         self.mm = BertForMaskedLM.from_pretrained(model)