-> style

1 files changed, 4 insertions, 3 deletions

diff --git a/utils.py b/utils.py
index 27ef833..465b0dc 100644
--- a/utils.py
+++ b/utils.py
@@ -122,7 +122,8 @@ def get_closest_to_point(one_embed, style_limit=[]):
 
 
 
-def translate_to_attr(embedding, to_attr, amt):
+def translate_to_attr(beer_id, to_attr, amt):
+    embedding = final_data[beer_id]['embed']
     if amt < 0:
         back = True
         amt = abs(amt)
@@ -157,8 +158,8 @@ def translate_to_attr(embedding, to_attr, amt):
     return get_closest_to_point(new_point)
 
 
-def translate_to_style(embedding, style):
-
+def translate_to_style(beer_id, style):
+    embedding = final_data[beer_id]['embed']
     closest_center_vector = style_centers[style]
     
     vector_between = closest_center_vector - embedding