Skip to content

Commit 55120b6

Browse files
author
Balaji Veeramani
committed
Fix integration tests
1 parent 7debe5b commit 55120b6

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

tests/integ/test_byo_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class _FactorizationMachineSerializer(BaseSerializer):
4040

4141
CONTENT_TYPE = "application/json"
4242

43-
def serialize(data):
43+
def serialize(self, data):
4444
js = {"instances": []}
4545
for row in data:
4646
js["instances"].append({"features": row.tolist()})

tests/integ/test_tuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -916,7 +916,7 @@ class _FactorizationMachineSerializer(BaseSerializer):
916916

917917
CONTENT_TYPE = "application/json"
918918

919-
def serializer(data):
919+
def serialize(self, data):
920920
js = {"instances": []}
921921
for row in data:
922922
js["instances"].append({"features": row.tolist()})

tests/integ/test_tuner_multi_algo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ class PredictionDataSerializer(BaseSerializer):
223223

224224
CONTENT_TYPE = "application/json"
225225

226-
def serialize(data):
226+
def serialize(self, data):
227227
js = {"instances": []}
228228
for row in data:
229229
js["instances"].append({"features": row.tolist()})

0 commit comments

Comments
 (0)