Merge pull request facebookresearch#248 from hpasapp/fix-object-array…

…-load fix error 'ValueError: Object arrays cannot be loaded when allow_pick…
duyphongdn1997 · May 25, 2020 · 3fe64a5 · 3fe64a5
2 parents 96f343c + aa91573
commit 3fe64a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/drqa/retriever/utils.py b/drqa/retriever/utils.py
@@ -30,7 +30,7 @@ def save_sparse_csr(filename, matrix, metadata=None):
 
 
 def load_sparse_csr(filename):
-    loader = np.load(filename)
+    loader = np.load(filename, allow_pickle=True)
     matrix = sp.csr_matrix((loader['data'], loader['indices'],
                             loader['indptr']), shape=loader['shape'])
     return matrix, loader['metadata'].item(0) if 'metadata' in loader else None