From d3f1a95c5f01c80b39400d70240fe9e4d90343b7 Mon Sep 17 00:00:00 2001 From: j Date: Mon, 3 Dec 2018 21:52:57 +0100 Subject: [PATCH] dont default to random and don't fail if it is missing --- recommendation_engine.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/recommendation_engine.py b/recommendation_engine.py index 64d7fe5..28bd228 100644 --- a/recommendation_engine.py +++ b/recommendation_engine.py @@ -62,8 +62,10 @@ class Engine: self.state['userKeywordsWeights'] = { 'themeTags': {'locked': False, 'value': 0.3}, 'characterTags': {'locked': False, 'value': 0.7}, - 'random' : {'locked': False, 'value': True} + 'random': {'locked': False, 'value': False} } + if 'random' not in self.state['userKeywordsWeights']: + self.state['userKeywordsWeights']['random'] = {'locked': False, 'value': False} self.update_keywords() @property @@ -263,7 +265,7 @@ class Engine: playlists.remove(playlist) # Generate random weights if random option is chosen in the dashboard: - if userKeywordsWeights['random']: + if userKeywordsWeights.get('random'): themeWeights = random.random() charWeights = 1-themeWeights else: