Skip to content

Commit 6bda57c

Browse files
committed
Update parallel tests to v11 (coclustering)
1 parent 28c8b28 commit 6bda57c

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tests/test_parallel_execution.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,10 @@ def test_parallel_coclustering_fit(self):
108108
callback=partial(
109109
KhiopsTestHelper.fit_helper,
110110
KhiopsCoclustering,
111-
key="SampleId",
112-
variables=["SampleId", "Pos", "Char"],
111+
fit_kwargs={
112+
"id_column": "SampleId",
113+
"columns": ["SampleId", "Pos", "Char"],
114+
},
113115
),
114116
arg_sequence=(secondary_train_data,) * n_clusterers,
115117
n_cpus=self._n_cpus,
@@ -201,8 +203,10 @@ def test_parallel_coclustering_fit_predict(self):
201203
callback=partial(
202204
KhiopsTestHelper.fit_helper,
203205
KhiopsCoclustering,
204-
key="SampleId",
205-
variables=["SampleId", "Pos", "Char"],
206+
fit_kwargs={
207+
"id_column": "SampleId",
208+
"columns": ["SampleId", "Pos", "Char"],
209+
},
206210
),
207211
arg_sequence=(secondary_train_data,) * n_clusterers,
208212
n_cpus=self._n_cpus,

0 commit comments

Comments
 (0)