diff --git a/tests/__init__.py b/pyengine/engine/faiss_wrapper/__init__.py similarity index 100% rename from tests/__init__.py rename to pyengine/engine/faiss_wrapper/__init__.py diff --git a/pyengine/engine/faiss_wrapper/faiss_wrapper.py b/pyengine/engine/faiss_wrapper/faiss_wrapper.py new file mode 100644 index 0000000000..6a60e53d35 --- /dev/null +++ b/pyengine/engine/faiss_wrapper/faiss_wrapper.py @@ -0,0 +1,28 @@ +import faiss +import numpy as np + +class FaissSearch(): + def __init__(self, index, id_to_vector_map): + pass + + # def search_by_ids(self, id_list, k): + # pass + + def search_by_vectors(self, vector_list, k): + # return both id and vector + pass + + def __search__(self, id_list, vector_list, k): + pass + + +class FaissIndex(): + def build_index(self, vector_list, dimension): + # return index + pass + + # def build_index_cpu(self): + # pass + + # def build_index_gpu(self): + # pass \ No newline at end of file diff --git a/pyengine/tests/__init__.py b/pyengine/tests/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/tests/basic_test.py b/pyengine/tests/basic_test.py similarity index 100% rename from tests/basic_test.py rename to pyengine/tests/basic_test.py