Fix rerank (#60)

* merge conflicts * merge conflicts * merge conflicts
veekaybee · Jul 12, 2023 · c635492 · c635492
1 parent 8611d32
commit c635492
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 6 deletions.
diff --git a/Makefile b/Makefile
@@ -16,7 +16,6 @@ test:
 embed:
 	docker exec -it viberary-flask-1 python /viberary/src/index/index_embeddings.py
 
-
 build:
 	docker compose build
 

diff --git a/src/index/index_embeddings.py b/src/index/index_embeddings.py
@@ -8,6 +8,7 @@
 conf = config()
 embedding_data = Path(conf["training_data"]["path"]) / Path(conf["training_data"]["file"])
 
+
 # Instantiate indexer
 indexer = Indexer(
     RedisConnection().conn(),

diff --git a/src/search/knn_search.py b/src/search/knn_search.py
@@ -96,7 +96,6 @@ def rescore(self, result_list: List[Tuple[float, str, str, str, int]]) -> List:
             for index, val in enumerate(result_list, 1)
         ]
 
-
     def dedup_by_number_of_reviews(
         self, result_list: List[Tuple[float, str, str, str, int]]
     ) -> List[Tuple[float, str, str, str, int]]:
@@ -105,13 +104,11 @@ def dedup_by_number_of_reviews(
         Args:
             result_list ():
 
-
         Returns: Deduped list by title and number of reviews
         """
 
         deduped_list = []
 
-
         titles = set()
 
         for element in result_list:
@@ -121,5 +118,4 @@ def dedup_by_number_of_reviews(
             else:
                 pass
 
-
         return deduped_list[0:10]
diff --git a/src/search/tests/test_ranker.py b/src/search/tests/test_ranker.py
@@ -48,7 +48,6 @@ def test_dedup(redis_mock):
     ]
 
     rescore = KNNSearch(redis_mock).dedup_by_number_of_reviews(result_list)
-    print(rescore)
     assert rescore == expected_list