osanseviero commited on
Commit
81a3c2a
1 Parent(s): 3337848

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -18,4 +18,4 @@ class PreTrainedPipeline():
18
  Return:
19
  A :obj:`list` of floats: The features computed by the model.
20
  """
21
- return self.model.get_sentence_vector(inputs)
 
18
  Return:
19
  A :obj:`list` of floats: The features computed by the model.
20
  """
21
+ return self.model.get_sentence_vector(inputs).tolist()