fix import (#10103)
This commit is contained in:
Родитель
4cda2d73ef
Коммит
226973a9c5
|
@ -144,7 +144,11 @@ try:
|
|||
_faiss_version = importlib_metadata.version("faiss")
|
||||
logger.debug(f"Successfully imported faiss version {_faiss_version}")
|
||||
except importlib_metadata.PackageNotFoundError:
|
||||
_faiss_available = False
|
||||
try:
|
||||
_faiss_version = importlib_metadata.version("faiss-cpu")
|
||||
logger.debug(f"Successfully imported faiss version {_faiss_version}")
|
||||
except importlib_metadata.PackageNotFoundError:
|
||||
_faiss_available = False
|
||||
|
||||
|
||||
_scatter_available = importlib.util.find_spec("torch_scatter") is not None
|
||||
|
|
Загрузка…
Ссылка в новой задаче