yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-25 16:15:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove conditional import of FAISS for Windows (#819)

Browse Source
This commit is contained in:
Tanay Soni 2021-02-12 12:15:23 +01:00 committed by GitHub
parent a1983ad84e
commit 8b0031bfc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
haystack/document_store/faiss.py
View File

@ -1,8 +1,8 @@
import logging
from sys import platform
from pathlib import Path
from typing import Union, List, Optional, Dict, Generator
from tqdm import tqdm
import faiss
import numpy as np
from haystack import Document
@ -11,10 +11,6 @@ from haystack.retriever.base import BaseRetriever
from haystack.utils import get_batches_from_generator
from scipy.special import expit
if platform != 'win32' and platform != 'cygwin':
import faiss
else:
raise ModuleNotFoundError("FAISSDocumentStore on windows platform is not supported")
logger = logging.getLogger(__name__)

2
requirements.txt
View File

@ -16,7 +16,7 @@ python-multipart
python-docx
sqlalchemy_utils
# for using FAISS with GPUs, install faiss-gpu
faiss-cpu==1.6.3; sys_platform != 'win32' and sys_platform != 'cygwin'
faiss-cpu==1.6.3
tika
uvloop==0.14; sys_platform != 'win32' and sys_platform != 'cygwin'
httptools