Merge branch 'coeff_parsing_support' of github.com:flatironinstitute/…

…generalized-linear-models into coeff_parsing_support
flatironinstitute · Nov 5, 2024 · 19ee425 · 19ee425
2 parents ef70fcb + 20bf0e7
commit 19ee425
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nemos/identifiability_constraints.py b/src/nemos/identifiability_constraints.py
@@ -129,7 +129,7 @@ def _add_invalid_entries(feature_matrix, shape_first_axis, is_valid):
 def _apply_identifiability_constraints(
     feature_matrix: JaxArray,
     preprocessing_func: Callable = add_constant,
-    warn_if_float32=True,
+    warn_if_float32: bool = True,
 ) -> Tuple[JaxArray, JaxArray]:
     """
     Apply identifiability constraints to a design matrix `feature_matrix`.