fixed comments from 10/25 + tox

flatironinstitute · Oct 26, 2024 · 12a3b16 · 12a3b16
1 parent d4af340
commit 12a3b16
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/nemos/glm.py b/src/nemos/glm.py
@@ -659,7 +659,7 @@ def fit(
         -------
         >>> # example input
         >>> import numpy as np
-        >>> X, y = np.random.normal(size=(10, 2)), np.random.uniform(size=10)
+        >>> X, y = np.random.normal(size=(10, 2)), np.random.poisson(size=10)
 
         >>> # fit a ridge regression Poisson GLM
         >>> import nemos as nmo
@@ -767,7 +767,7 @@ def simulate(
         --------
         >>> # example input
         >>> import numpy as np
-        >>> X, y = np.random.normal(size=(10, 2)), np.random.uniform(size=10)
+        >>> X, y = np.random.normal(size=(10, 2)), np.random.poisson(size=10)
 
         >>> # define and fit model
         >>> import nemos as nmo
@@ -956,7 +956,7 @@ def initialize_state(
         --------
         >>> import numpy as np
         >>> import nemos as nmo
-        >>> X, y = np.random.normal(size=(10, 2)), np.random.uniform(size=10)
+        >>> X, y = np.random.normal(size=(10, 2)), np.random.poisson(size=10)
         >>> model = nmo.glm.GLM()
         >>> params = model.initialize_params(X, y)
         >>> opt_state = model.initialize_state(X, y, params)