Merge pull request #710 from QData/format

format after #695
QData · Dec 21, 2022 · edbcb83 · edbcb83
2 parents 8a7c88d + 7c152d9
commit edbcb83
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 7 deletions.
diff --git a/textattack/metrics/quality_metrics/bert_score.py b/textattack/metrics/quality_metrics/bert_score.py
@@ -13,7 +13,9 @@
 
 class BERTScoreMetric(Metric):
  def __init__(self, **kwargs):
- self.use_obj = BERTScore(min_bert_score=0.5, model_name="microsoft/deberta-large-mnli", num_layers=18)
+ self.use_obj = BERTScore(
+ min_bert_score=0.5, model_name="microsoft/deberta-large-mnli", num_layers=18
+ )
  self.original_candidates = []
  self.successful_candidates = []
  self.all_metrics = {}

diff --git a/textattack/metrics/quality_metrics/meteor_score.py b/textattack/metrics/quality_metrics/meteor_score.py
@@ -6,8 +6,9 @@
 
 """
 
-from textattack.attack_results import FailedAttackResult, SkippedAttackResult
 import nltk
+
+from textattack.attack_results import FailedAttackResult, SkippedAttackResult
 from textattack.metrics import Metric
 
 
@@ -54,13 +55,20 @@ def calculate(self, results):
  elif isinstance(result, SkippedAttackResult):
  continue
  else:
- self.original_candidates.append(result.original_result.attacked_text.text)
- self.successful_candidates.append(result.perturbed_result.attacked_text.text)
+ self.original_candidates.append(
+ result.original_result.attacked_text.text
+ )
+ self.successful_candidates.append(
+ result.perturbed_result.attacked_text.text
+ )
 
  meteor_scores = []
  for c in range(len(self.original_candidates)):
  meteor_scores.append(
- nltk.translate.meteor([nltk.word_tokenize(self.original_candidates[c])], nltk.word_tokenize(self.successful_candidates[c]))
+ nltk.translate.meteor(
+ [nltk.word_tokenize(self.original_candidates[c])],
+ nltk.word_tokenize(self.successful_candidates[c]),
+ )
  )
 
  self.all_metrics["avg_attack_meteor_score"] = round(

diff --git a/textattack/metrics/quality_metrics/sentence_bert.py b/textattack/metrics/quality_metrics/sentence_bert.py
@@ -13,13 +13,14 @@
 
 class SBERTMetric(Metric):
  def __init__(self, **kwargs):
- self.use_obj = BERT(model_name="all-MiniLM-L6-v2",  metric="cosine")
+ self.use_obj = BERT(model_name="all-MiniLM-L6-v2", metric="cosine")
  self.original_candidates = []
  self.successful_candidates = []
  self.all_metrics = {}
 
  def calculate(self, results):
- """Calculates average Sentence BERT similarity on all successfull attacks.
+ """Calculates average Sentence BERT similarity on all successfull
+ attacks.
 
  Args:
  results (``AttackResult`` objects):