grantpitt commited on
Commit
c7c178b
1 Parent(s): 1cfb47e

need to return a list, fixed

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -35,4 +35,4 @@ class PreTrainedPipeline():
35
  n = 50
36
  top_sign_ids = self.sign_ids[sign_id_arg_rankings[:n]]
37
  top_sign_similarities = cos_similarites[sign_id_arg_rankings[:n]]
38
- return dict(zip(top_sign_ids.tolist(), top_sign_similarities.tolist()))
 
35
  n = 50
36
  top_sign_ids = self.sign_ids[sign_id_arg_rankings[:n]]
37
  top_sign_similarities = cos_similarites[sign_id_arg_rankings[:n]]
38
+ return [top_sign_ids.tolist(), top_sign_similarities.tolist()]