aynetdia commited on
Commit
6fc5137
·
1 Parent(s): a8d5201
Files changed (2) hide show
  1. semscore.py +1 -1
  2. tests.py +0 -17
semscore.py CHANGED
@@ -84,7 +84,7 @@ class SemScore(evaluate.Metric):
84
 
85
  def _download_and_prepare(self, dl_manager):
86
  """Optional: download external resources useful to compute the scores"""
87
- if self.config_name is None:
88
  checkpoint = "sentence-transformers/all-mpnet-base-v2"
89
  else:
90
  checkpoint = self.config_name
 
84
 
85
  def _download_and_prepare(self, dl_manager):
86
  """Optional: download external resources useful to compute the scores"""
87
+ if self.config_name == "default":
88
  checkpoint = "sentence-transformers/all-mpnet-base-v2"
89
  else:
90
  checkpoint = self.config_name
tests.py DELETED
@@ -1,17 +0,0 @@
1
- test_cases = [
2
- {
3
- "predictions": [0, 0],
4
- "references": [1, 1],
5
- "result": {"metric_score": 0}
6
- },
7
- {
8
- "predictions": [1, 1],
9
- "references": [1, 1],
10
- "result": {"metric_score": 1}
11
- },
12
- {
13
- "predictions": [1, 0],
14
- "references": [1, 1],
15
- "result": {"metric_score": 0.5}
16
- }
17
- ]