Skip to content

Commit d71990d

Browse files
committed
test fix
1 parent f317bf0 commit d71990d

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/test_weighting/test_ntk_weighting.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,11 @@ def test_constructor(model, alpha):
2424
NeuralTangetKernelWeighting(model=model, alpha=alpha)
2525

2626

27+
@pytest.mark.parametrize("model", [(0.5)])
28+
def test_wrong_constructor1(model):
29+
with pytest.raises(TypeError):
30+
NeuralTangetKernelWeighting(model)
31+
2732
@pytest.mark.parametrize(
2833
"model,alpha",
2934
[
@@ -40,10 +45,6 @@ def test_wrong_constructor2(model, alpha):
4045
NeuralTangetKernelWeighting(model, alpha)
4146

4247

43-
@pytest.mark.parametrize("model,alpha", [(0.5, 0.5)])
44-
def test_wrong_constructor1(model):
45-
with pytest.raises(TypeError):
46-
NeuralTangetKernelWeighting(model)
4748

4849

4950
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)