diff --git a/recommendation_engine.py b/recommendation_engine.py index 0fdf31e..8d0fa4b 100644 --- a/recommendation_engine.py +++ b/recommendation_engine.py @@ -137,10 +137,10 @@ class Engine: next_playlist_index.append(i) break # Discuss how this behavour should be: should it switch to a new playlist if it is the end of the playlist clip sequence already? - elif prev_grid_list[i].get('playlistPosition') + 1 == len(playlist['clips']): + elif prev_grid_list[i].get('playlistPosition', 0) + 1 == len(playlist['clips']): playlist_pos = 0 else: - playlist_pos = prev_grid_list[i].get('playlistPosition') + 1 + playlist_pos = prev_grid_list[i].get('playlistPosition', 0) + 1 rec_list.append((i, { 'clips': playlist['clips'],