Wendaxu commited on
Commit
578bca9
1 Parent(s): cb87d9f

remove src

Browse files
Files changed (1) hide show
  1. sescore.py +1 -1
sescore.py CHANGED
@@ -134,7 +134,7 @@ class SEScore(evaluate.Metric):
134
  destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
135
  self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
136
 
137
- def _compute(self, sources, predictions, references, gpus=None, progress_bar=False):
138
  if gpus is None:
139
  gpus = 1 if torch.cuda.is_available() else 0
140
  data = {"src": references, "mt": predictions}
134
  destination = snapshot_download(repo_id="xu1998hz/sescore_english_mt", revision="main")
135
  self.scorer = load_from_checkpoint(f'{destination}/checkpoint/sescore_english_mt.ckpt')
136
 
137
+ def _compute(self, predictions, references, gpus=None, progress_bar=False):
138
  if gpus is None:
139
  gpus = 1 if torch.cuda.is_available() else 0
140
  data = {"src": references, "mt": predictions}