Update logic.py

logic.py

@@ -404,6 +404,7 @@ def make_pyg_graph(movie_rec_db):
     encoded_genres = GenresEncoder(movie_rec_db.aql.execute('FOR doc IN Movie RETURN doc'))
     movie_x = torch.cat((title_emb, encoded_genres), dim=-1)
 
+    global data
     data = HeteroData()
     data['user'].num_nodes = len(users)  # Users do not have any features.
     data['movie'].x = movie_x