Merge pull request #257 from sadimanna/main

updated SentEval/senteval/sts.py
princeton-nlp · Jul 2, 2024 · 7edb07e · 7edb07e
2 parents 13361d0 + ec1494c
commit 7edb07e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/SentEval/senteval/sts.py b/SentEval/senteval/sts.py
@@ -39,8 +39,8 @@ def loadFile(self, fpath):
             not_empty_idx = raw_scores != ''
 
             gs_scores = [float(x) for x in raw_scores[not_empty_idx]]
-            sent1 = np.array([s.split() for s in sent1])[not_empty_idx]
-            sent2 = np.array([s.split() for s in sent2])[not_empty_idx]
+            sent1 = np.array([s.split() for s in sent1], dtype=object)[not_empty_idx]
+            sent2 = np.array([s.split() for s in sent2], dtype=object)[not_empty_idx]
             # sort data by length to minimize padding in batcher
             sorted_data = sorted(zip(sent1, sent2, gs_scores),
                                  key=lambda z: (len(z[0]), len(z[1]), z[2]))