add seed to test

bambinos · Jan 6, 2023 · 840172a · 840172a
1 parent afd9e6a
commit 840172a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_glmlss.py b/tests/test_glmlss.py
@@ -56,13 +56,13 @@ def test_gamma(data_gamma, data_new_gamma):
 
     # Default links
     model = Model(formula, data_gamma, family="gamma")
-    idata = model.fit(tune=100, draws=100)
+    idata = model.fit(tune=100, draws=100, random_seed=1234)
     model.predict(idata, kind="pps")
     model.predict(idata, kind="pps", data=data_new_gamma)
 
     # Custom links
     model = Model(formula, data_gamma, family="gamma", link={"mu": "log", "alpha": "log"})
-    idata = model.fit(tune=100, draws=100)
+    idata = model.fit(tune=100, draws=100, random_seed=1234)
     model.predict(idata, kind="pps")
     model.predict(idata, kind="pps", data=data_new_gamma)
 
@@ -72,6 +72,6 @@ def test_gamma_with_splines(data_normal, data_new_normal):
         "count ~ 0 + bs(hour, 8, intercept=True)", "alpha ~ 0 + bs(hour, 8, intercept=True)"
     )
     model = Model(formula, data_normal, family="gamma", link={"mu": "log", "alpha": "log"})
-    idata = model.fit(tune=100, draws=100)
+    idata = model.fit(tune=100, draws=100, random_seed=1234)
     model.predict(idata, kind="pps")
     model.predict(idata, kind="pps", data=data_new_normal)