Add the ability to read from .npy file (#14)

Author: slice4e

bindings/python/src/pysvs/__init__.py

@@ -19,5 +19,5 @@
 )
 
 # Misc types and functions
-from .common import read_vecs, write_vecs, read_svs, k_recall_at, generate_test_dataset
+from .common import read_vecs, read_npy, write_vecs, read_svs, k_recall_at, generate_test_dataset
 

bindings/python/src/pysvs/common.py

@@ -18,6 +18,21 @@
 from .loader import library
 lib = library()
 
+def read_npy(filename:str):
+    """
+    Read a file in the `npy` format and return a NumPy array with the results.
+
+    Args:
+        filename: The file to read.
+
+    Returns:
+        Numpy array with the results.
+    """
+
+    X = np.load(filename)
+    return np.ascontiguousarray(X)
+
+
 def read_vecs(filename: str):
     """
     Read a file in the `bvecs/fvecs/ivecs` format and return a NumPy array with the results.