Update gps_classifier.py

fix for ronikobrosly#45
dirknbr · May 21, 2024 · 31b909e · 31b909e
1 parent a06fd68
commit 31b909e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/causal_curve/gps_classifier.py b/causal_curve/gps_classifier.py
@@ -121,6 +121,6 @@ def _create_log_odds(self, T):
         """
         return logit(
             self.gam_results.predict_proba(
-                np.array([T, self.gps_function(T).mean()]).reshape(1, -1)
+                np.array([T[0], self.gps_function(T).mean()]).reshape(1, -1)
             )
         )