Bug fix

famura · Jul 9, 2024 · 9f39ac5 · 9f39ac5
1 parent 0cd4d81
commit 9f39ac5
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_neural_fields.py b/tests/test_neural_fields.py
@@ -21,7 +21,7 @@
 @pytest.mark.parametrize("conv_out_channels", [1, 5], ids=["1dim_conv_out", "5dim_conv_out"])
 @pytest.mark.parametrize("tau_init", [10], ids=["default_tau_init"])
 @pytest.mark.parametrize("tau_learnable", [True, False], ids=["learnable_tau", "fixed_tau"])
-@pytest.mark.parametrize("kappa_init", [0, 1e-2], ids=["kappa_init_0", "1e-2"])
+@pytest.mark.parametrize("kappa_init", [1e-2], ids=["kappa_1e-2"])
 @pytest.mark.parametrize("kappa_learnable", [True, False], ids=["learnable_kappa", "fixed_kappa"])
 @pytest.mark.parametrize("potentials_init", [None], ids=["default_pot_init"])
 @pytest.mark.parametrize("init_param_kwargs", [None, dict(bell=True)], ids=["default_init", "bell_init"])

diff --git a/tests/test_simple_neural_fields.py b/tests/test_simple_neural_fields.py
@@ -32,7 +32,7 @@
 @pytest.mark.parametrize("activation_nonlin", [torch.sigmoid, torch.tanh], ids=["sigmoid", "tanh"])
 @pytest.mark.parametrize("tau_init", [10], ids=["default_tau_init"])
 @pytest.mark.parametrize("tau_learnable", [True, False], ids=["learnable_tau", "fixed_tau"])
-@pytest.mark.parametrize("kappa_init", [0, 1e-2], ids=["kappa_init_0", "1e-2"])
+@pytest.mark.parametrize("kappa_init", [1e-2], ids=["kappa_1e-2"])
 @pytest.mark.parametrize("kappa_learnable", [True, False], ids=["learnable_kappa", "fixed_kappa"])
 @pytest.mark.parametrize("capacity_learnable", [True, False], ids=["learnable_capacity", "fixed_capacity"])
 @pytest.mark.parametrize("potentials_init", [None], ids=["default_pot_init"])