Importing faiss.swigfaiss leads to error

[pritam-dey3]

Describe the bug I was trying to import faiss.swigfaiss.Index to static type index in my code. But it took me quite some painful hours to figure out the issue.

To Reproduce The following code works.

import faiss
# from faiss.swigfaiss import Index
import numpy as np

index = faiss.index_factory(16, "IDMap,Flat")
emb = np.random.uniform(0, 1, (8, 16)).astype(np.float32)

index.train(emb)

however, the following code fails

import faiss
from faiss.swigfaiss import Index
import numpy as np

index = faiss.index_factory(16, "IDMap,Flat")
emb = np.random.uniform(0, 1, (8, 16)).astype(np.float32)

index.train(emb)

  File "<string>", line 9, in <module>
TypeError: IndexIDMap.train() missing 1 required positional argument: 'x'

In conda installation, both the above code works fine.

Desktop (please complete the following information):

• OS: Windows 11
• Architecture: x86_64
• Python: 3.12.4
• Version: 1.8.0

[kyamagu]

Hmm, this seems not to happen in the following:

import faiss
from faiss.swigfaiss_avx2 import Index
import numpy as np

index = faiss.index_factory(16, "IDMap,Flat")
emb = np.random.uniform(0, 1, (8, 16)).astype(np.float32)
index.train(emb)

This indicates something is wrong when the optimization flag is not set

[kyamagu]

Btw, it is possible to import Index in the following way:

import faiss
from faiss import Index