Skip to content

Commit 4fbe701

Browse files
committed
fix test_check_basis_with_layer
1 parent 907f824 commit 4fbe701

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

n3fit/src/n3fit/tests/test_checks.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,13 @@ def test_check_basis_with_layer():
4747
basis = [{"fl": i} for i in flavs]
4848
layers = [4, 5, 9]
4949
vp_basis = check_basis("flavour", flavs)["basis"]
50+
trial_specs = {}
5051
with pytest.raises(CheckError):
51-
checks.check_basis_with_layers(basis, vp_basis, {"nodes_per_layer": layers})
52-
# Or when the wrong kind of basis is veing used
52+
checks.check_basis_with_layers(basis, vp_basis, {"nodes_per_layer": layers}, trial_specs)
53+
# Or when the wrong kind of basis is being used
5354
params = {"nodes_per_layer": [4, 5], "activation_per_layer": ["sigmoid", "square_singlet"]}
5455
with pytest.raises(CheckError):
55-
checks.check_basis_with_layers(basis, vp_basis, params)
56+
checks.check_basis_with_layers(basis, vp_basis, params, trial_specs, trial_specs)
5657

5758

5859
def test_check_optimizer():

0 commit comments

Comments
 (0)