fixed tests

flatironinstitute · Oct 21, 2024 · 8a08d3a · 8a08d3a
1 parent 5ca9b49
commit 8a08d3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_glm.py b/tests/test_glm.py
@@ -891,7 +891,7 @@ def test_initialize_solver_param_length(
             model.initialize_state(X, y, params)
 
     @pytest.mark.parametrize(
-        "inv_link", [jnp.exp, lambda x: jnp.exp(x), jax.nn.softplus, jax.nn.sigmoid]
+        "inv_link", [jnp.exp, lambda x: jnp.exp(x), jax.nn.softplus, jax.nn.relu]
     )
     def test_high_firing_rate_initialization(
         self, inv_link, example_X_y_high_firing_rates
@@ -1932,7 +1932,7 @@ def test_init_observation_type(
             )
 
     @pytest.mark.parametrize(
-        "inv_link", [jnp.exp, lambda x: jnp.exp(x), jax.nn.softplus, jax.nn.sigmoid]
+        "inv_link", [jnp.exp, lambda x: jnp.exp(x), jax.nn.softplus, jax.nn.relu]
     )
     def test_high_firing_rate_initialization(
         self, inv_link, example_X_y_high_firing_rates