Update src/nemos/identifiability_constraints.py

Co-authored-by: William F. Broderick <[email protected]>
flatironinstitute · Nov 5, 2024 · 20bf0e7 · 20bf0e7
1 parent ac341f7
commit 20bf0e7
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`.