From 2a77fb8d847f27b8f68156acceb23df0cdbe5059 Mon Sep 17 00:00:00 2001
From: Ben Cohen <ben@kensho.com>
Date: Sun, 28 Jul 2019 15:31:44 -0400
Subject: -> style

---
 utils.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'utils.py')

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
-- 
cgit v1.2.3