diff --git a/db2pc.py b/db2pc.py index 557462e..caf0ae7 100644 --- a/db2pc.py +++ b/db2pc.py @@ -32,7 +32,7 @@ for x in tqdm(range(0,len(df),batch_size)): for idx, value in enumerate(trakt_ids): to_send.append( { - value: embeddings[idx] + value: embeddings[idx].tolist() }) print(to_send)