Skip to content

Commit a93f9d0

Browse files
committed
more unit test updating
1 parent 59863ac commit a93f9d0

4 files changed

Lines changed: 4 additions & 10 deletions

File tree

‎tests/conftest.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -532,9 +532,7 @@ def sklearn_classification_model(iris_dataset):
532532

533533
with warnings.catch_warnings():
534534
warnings.simplefilter("ignore")
535-
model = sk.LogisticRegression(
536-
multi_class="multinomial", solver="lbfgs", max_iter=1000
537-
)
535+
model = sk.LogisticRegression(solver="lbfgs", max_iter=1000)
538536
model.fit(X, y)
539537
return model
540538

‎tests/integration/test_model_parameters.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def sklearn_model(train_data):
4242
X, y = train_data
4343
with warnings.catch_warnings():
4444
warnings.simplefilter("ignore")
45-
model = LogisticRegression(multi_class="multinomial", solver="lbfgs")
45+
model = LogisticRegression(solver="lbfgs")
4646
model.fit(X, y)
4747
return model
4848

‎tests/integration/test_pymas.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,7 @@ def sklearn_model(train_data):
6363
X, y = train_data
6464
with warnings.catch_warnings():
6565
warnings.simplefilter("ignore")
66-
model = LogisticRegression(
67-
multi_class="multinomial", solver="lbfgs", max_iter=10000
68-
)
66+
model = LogisticRegression(solver="lbfgs", max_iter=10000)
6967
model.fit(X, y)
7068
return model
7169

‎tests/integration/test_tasks.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,7 @@ def sklearn_logistic_model():
4040

4141
with warnings.catch_warnings():
4242
warnings.simplefilter("ignore")
43-
model = LogisticRegression(
44-
multi_class="multinomial", solver="lbfgs", max_iter=10000
45-
)
43+
model = LogisticRegression(solver="lbfgs", max_iter=10000)
4644
model.fit(iris.iloc[:, 0:4], iris["Species"])
4745

4846
return model, iris.iloc[:, 0:4]

0 commit comments

Comments
 (0)