Skip to content

Commit 19d9f8c

Browse files
committed
fix: convert X to numpy.array in glove dataset function
1 parent cc3a748 commit 19d9f8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ann_benchmarks/datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def glove(out_fn: str, d: int) -> None:
213213
for line in z.open(z_fn):
214214
v = [float(x) for x in line.strip().split()[1:]]
215215
X.append(numpy.array(v))
216-
X_train, X_test = train_test_split(X)
216+
X_train, X_test = train_test_split(numpy.array(X))
217217
write_output(numpy.array(X_train), numpy.array(X_test), out_fn, "angular")
218218

219219

0 commit comments

Comments
 (0)