Convert 2D ndarray to list in most_similar

piskvorky · Feb 20, 2022 · 330e5c2 · 330e5c2
1 parent 2c041d0
commit 330e5c2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -203,6 +203,9 @@ def _ensure_list(value):
     if isinstance(value, _KEY_TYPES) or (isinstance(value, ndarray) and len(value.shape) == 1):
         return [value]
 
+    if isinstance(value, ndarray) and len(value.shape) == 2:
+        return list(value)
+
     return value