fix pyright errors

afsl/sift.py

@@ -3,7 +3,7 @@
 from afsl.acquisition_functions import AcquisitionFunction, Targeted
 from afsl.acquisition_functions.lazy_vtl import LazyVTL
 from afsl.acquisition_functions.vtl import VTL
-import faiss  # type: ignore
+import faiss
 import torch
 import time
 import concurrent.futures
@@ -46,12 +46,12 @@ class Retriever:
     ```
     """
 
-    index: faiss.Index  # type: ignore
+    index: faiss.Index
     only_faiss: bool = False
 
     def __init__(
         self,
-        index: faiss.Index,  # type: ignore
+        index: faiss.Index,
         acquisition_function: AcquisitionFunction | None = None,
         llambda: float = 0.01,
         fast: bool = False,
@@ -148,7 +148,7 @@ def batch_search(
         mean_queries = np.mean(queries, axis=1)
 
         t_start = time.time()
-        faiss.omp_set_num_threads(threads)  # type: ignore
+        faiss.omp_set_num_threads(threads)
         D, I, V = self.index.search_and_reconstruct(mean_queries, K or self.index.ntotal)  # type: ignore
         t_faiss = time.time() - t_start
 

examples/fine_tuning/training.py

@@ -91,10 +91,8 @@ def train_loop(
     if faiss_index_path is not None:
         acquisition_function.subsample = False
 
-        res = faiss.StandardGpuResources()
         index = faiss.read_index(faiss_index_path)
-        index = faiss.index_cpu_to_gpu(res, 0, index)
-        retriever = Retriever(index, acquisition_function)  # type: ignore
+        retriever = Retriever(index, acquisition_function)
     else:
         data_loader = ActiveDataLoader(
             dataset=train_inputs,

requirements.txt

@@ -1,4 +1,5 @@
 black~=24.8
+faiss-cpu~=1.8
 numpy~=2.1
 pandas~=2.2
 poetry~=1.8