Skip to content

Commit 9813fda

Browse files
Remove cluster_type argument from Distributed in test_forest.py
1 parent a73cddd commit 9813fda

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/systems/ops/fil/test_forest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_export_merlin_models(tmpdir):
8787

8888
# train a XGB model using merlin-models
8989
model = merlin_xgb.XGBoost(ds.schema)
90-
with Distributed(cluster_type="cpu"):
90+
with Distributed():
9191
model.fit(ds)
9292

9393
# make sure we can export the merlin-model xgb wrapper using merlin-systems

0 commit comments

Comments
 (0)