need to return a list, fixed
Browse files- 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
|
|
|
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()]
|