Skip to content

Commit

Permalink
pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
vprelovac committed Oct 25, 2023
1 parent bb0225a commit 72f04d9
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions vectordb/vector_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,13 @@

#pylint: disable = line-too-long, trailing-whitespace, trailing-newlines, line-too-long, missing-module-docstring, import-error, too-few-public-methods, too-many-instance-attributes, too-many-locals

from typing import List
from typing import List, Tuple
import numpy as np
import faiss
import sklearn



MRPT_LOADED = True
try:
import mrpt
Expand Down Expand Up @@ -61,7 +63,7 @@ def run_sk(vector, vectors, k=15):
@staticmethod
def search_vectors(
query_embedding: List[float], embeddings: List[List[float]], top_n: int
) -> List[tuple[int, float]]:
) -> List[Tuple[int, float]]:
"""
Searches for the most similar vectors to the query_embedding in the given embeddings.
Expand Down

0 comments on commit 72f04d9

Please sign in to comment.