Merge branch 'development' into set_shape_basis_method

flatironinstitute · Dec 18, 2024 · b5b5e3f · b5b5e3f
2 parents 2ade4bd + 2682aac
commit b5b5e3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nemos/initialize_regressor.py b/src/nemos/initialize_regressor.py
@@ -81,7 +81,7 @@ def initialize_intercept_matching_mean_rate(
     # return inverse if analytical solution is available
     analytical_inv = INVERSE_FUNCS.get(inverse_link_function, None)
 
-    means = jnp.atleast_1d(jnp.mean(y, axis=0))
+    means = jnp.atleast_1d(jnp.nanmean(y, axis=0))
     if analytical_inv:
         out = analytical_inv(means)
         if jnp.any(jnp.isnan(out)):