grantpitt commited on
Commit
96f94cc
1 Parent(s): 39152b4

I think you have to return something that's json serializable. I needed to convert the np array to list

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