Skip to content

Commit df5f14d

Browse files
committed
fix format issue in chinse_word_swap_masked.py
1 parent e8f74bc commit df5f14d

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,3 @@ more-itertools
2020
pinyin>=0.4.0
2121
jieba
2222
OpenHowNet
23-
click==8.0.2

textattack/transformations/word_swaps/chn_transformations/chinese_word_swap_masked.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
-------------------------------------
44
"""
55

6-
from transformers import pipeline
6+
import torch
77

88
from . import WordSwap
99

@@ -13,8 +13,7 @@ class ChineseWordSwapMaskedLM(WordSwap):
1313
model."""
1414

1515
def __init__(self, task="fill-mask", model="xlm-roberta-base", **kwargs):
16-
from transformers import BertTokenizer, BertForMaskedLM
17-
import torch
16+
from transformers import BertForMaskedLM, BertTokenizer
1817

1918
self.tt = BertTokenizer.from_pretrained(model)
2019
self.mm = BertForMaskedLM.from_pretrained(model)

0 commit comments

Comments
 (0)