From 4570d3e2fdf2cc58fd46431ee2a75e866781c481 Mon Sep 17 00:00:00 2001 From: Steven Dahdah Date: Fri, 16 Dec 2022 18:28:43 -0500 Subject: [PATCH] Remove reference to X_ikm1 --- pykoop/koopman_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pykoop/koopman_pipeline.py b/pykoop/koopman_pipeline.py index 9f8cd51..254381c 100644 --- a/pykoop/koopman_pipeline.py +++ b/pykoop/koopman_pipeline.py @@ -2564,7 +2564,8 @@ def predict_trajectory( ) X_i[[k + self.min_samples_ - 1], :] = X_ik[[-1], :] except ValueError as ve: - if (np.all(np.isfinite(X_ikm1)) + if (np.all(np.isfinite(X_ik)) + and np.all(np.isfinite(X_i)) and np.all(np.isfinite(Theta_i)) and np.all(np.isfinite(Upsilon_i)) and np.all(np.isfinite(U_i))):