Skip to content

Commit f16795d

Browse files
Krsto ProrokovićKrsto Proroković
authored andcommitted
Fix some unit tests
1 parent 0b8983c commit f16795d

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

test/scholar/linear/logistic_regression_test.exs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,8 @@ defmodule Scholar.Linear.LogisticRegressionTest do
4747
assert_raise NimbleOptions.ValidationError,
4848
"""
4949
invalid value for :optimizer option: expected :optimizer to be either \
50-
a valid 0-arity function in Polaris.Optimizers or a valid {init_fn, update_fn} tuple
50+
a valid 0-arity function in Polaris.Optimizers or a valid {init_fn, update_fn} tuple\
5151
""",
52-
# "invalid value for :optimizer option: expected :optimizer to be either a valid 0-arity function in Polaris.Optimizers or a valid {init_fn, update_fn} tuple",
5352
fn ->
5453
LogisticRegression.fit(x, y,
5554
num_classes: 2,
@@ -84,7 +83,7 @@ defmodule Scholar.Linear.LogisticRegressionTest do
8483

8584
assert_raise ArgumentError,
8685
"""
87-
Scholar.Linear.LogisticRegression expected y to have shape {num_samples}, \
86+
expected y to have shape {num_samples}, \
8887
got tensor with shape: {2, 2}
8988
""",
9089
fn -> LogisticRegression.fit(x, y, num_classes: 2) end

0 commit comments

Comments
 (0)