pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 faiss,
5 faiss-build,
6 scipy,
7 pytestCheckHook,
8}:
9
10assert faiss.pythonSupport;
11
12buildPythonPackage {
13 pname = "faiss-pytest-suite";
14 inherit (faiss) version;
15
16 format = "other";
17
18 src = "${faiss-build.src}/tests";
19
20 dontBuild = true;
21 dontInstall = true;
22
23 # Tests that need GPUs and would fail in the sandbox
24 disabledTestPaths = lib.optionals faiss.cudaSupport [ "test_contrib.py" ];
25
26 disabledTests = [
27 # https://github.com/facebookresearch/faiss/issues/2836
28 "test_update_codebooks_with_double"
29 ];
30
31 nativeCheckInputs = [
32 faiss
33 pytestCheckHook
34 scipy
35 ];
36
37 meta = faiss.meta // {
38 description = "Faiss test suite";
39 };
40}