From 20bf0e7607b6b494c49d53ec71852a4fcecf5a40 Mon Sep 17 00:00:00 2001
From: Edoardo Balzani <edoardo.balzani87@gmail.com>
Date: Tue, 5 Nov 2024 11:11:49 -0500
Subject: [PATCH] Update src/nemos/identifiability_constraints.py

Co-authored-by: William F. Broderick <billbrod@gmail.com>
---
 src/nemos/identifiability_constraints.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/nemos/identifiability_constraints.py b/src/nemos/identifiability_constraints.py
index 6e3ff46b..d261d383 100644
--- 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`.