shit

1 files changed, 4 insertions, 2 deletions

diff --git a/utils.py b/utils.py
index a9a0786..27ef833 100644
--- a/utils.py
+++ b/utils.py
@@ -88,12 +88,14 @@ for e in embeddings:
 def get_closest(beer_id):
     one_embed = final_data[beer_id]['embed']
     
-
+    beer_ids = []
     for thing in sorted(small_embeddings, key = lambda x: cosine(one_embed, x[1]), reverse=False)[:25]:
 
         if thing[0] in beer_names:
-            print(beer_names[thing[0]])
+            beer_ids.append(thing[0])
+            print(thing[0])
             print('=' * 50)
+    return beer_ids
             
 def get_closest_to_point(one_embed, style_limit=[]):
 #     one_embed = final_data[beer_id]['embed']