I think you have to return something that's json serializable. I needed to convert the np array to list
Browse files- 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()
|