update tests

bioAI-Oslo · Sep 15, 2023 · 2b1eb8b · 2b1eb8b
1 parent 9cca1e9
commit 2b1eb8b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -87,7 +87,7 @@ def bernoulli_glm():
     from spikeometric.models import BernoulliGLM
     rng = torch.Generator().manual_seed(14071789)
     model = BernoulliGLM(
-        theta=5.,
+        b=-5.,
         dt=1,
         coupling_window=5,
         abs_ref_scale=3,
@@ -96,6 +96,7 @@ def bernoulli_glm():
         rel_ref_scale=7,
         rel_ref_strength=-30.,
         alpha=0.2,
+        gain=1,
         rng=rng,
     )
     return model
@@ -105,7 +106,7 @@ def threshold_sam():
     from spikeometric.models import ThresholdSAM
     rng = torch.Generator().manual_seed(14071789)
     model = ThresholdSAM(
-        r=0.025,
+        gain=0.025,
         b=0.001,
         tau=10,
         dt=0.1,
@@ -126,7 +127,7 @@ def poisson_glm():
         dt=0.1,
         T=200,
         tau=10,
-        r=0.025,
+        gain=0.025,
         b=0.001,
         rng=rng,
     )
@@ -142,7 +143,7 @@ def rectified_lnp():
         dt=0.1,
         T=200,
         tau=10.,
-        r=0.025,
+        gain=0.025,
         b=0.001,
         rng=rng,
     )
@@ -157,7 +158,7 @@ def rectified_sam():
         theta=-1.37e-2,
         tau=10.,
         dt=0.1,
-        r=0.025,
+        gain=0.025,
         b=0.001,
         rng=rng,
     )