adds pinned faiss 1.6.1

Alex · Alex · commit b0d7a18946a8 · 2023-02-28T10:41:32.000-08:00
diff --git a/setup.py b/setup.py
@@ -40,7 +40,8 @@ def unique_flatten_dict(d):
 base_extras_heavy = {
   'umap-learn': ['umap-learn', 'dirty-cat==0.2.0', 'scikit-learn>=1.0'],
 }
-base_extras_heavy['ai'] = base_extras_heavy['umap-learn'] + ['setuptools==67.4.0', 'scipy', 'dgl', 'torch', 'sentence-transformers', 'faiss-cpu==1.6.5', 'joblib']
+# https://github.com/facebookresearch/faiss/issues/1589 for faiss-cpu 1.6.1, #'setuptools==67.4.0' removed
+base_extras_heavy['ai'] = base_extras_heavy['umap-learn'] + ['scipy', 'dgl', 'torch', 'sentence-transformers', 'faiss-cpu==1.6.1', 'joblib']
 
 base_extras = {**base_extras_light, **base_extras_heavy}
 

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,8 @@ def unique_flatten_dict(d):`
`40`	`40`	`base_extras_heavy = {`
`41`	`41`	`'umap-learn': ['umap-learn', 'dirty-cat==0.2.0', 'scikit-learn>=1.0'],`
`42`	`42`	`}`
`43`		`-base_extras_heavy['ai'] = base_extras_heavy['umap-learn'] + ['setuptools==67.4.0', 'scipy', 'dgl', 'torch', 'sentence-transformers', 'faiss-cpu==1.6.5', 'joblib']`
	`43`	`+# https://github.com/facebookresearch/faiss/issues/1589 for faiss-cpu 1.6.1, #'setuptools==67.4.0' removed`
	`44`	`+base_extras_heavy['ai'] = base_extras_heavy['umap-learn'] + ['scipy', 'dgl', 'torch', 'sentence-transformers', 'faiss-cpu==1.6.1', 'joblib']`
`44`	`45`
`45`	`46`	`base_extras = {base_extras_light, base_extras_heavy}`
`46`	`47`