From 8bfa8a03bb4396eae926164cd687094e65bf7d5f Mon Sep 17 00:00:00 2001 From: Alan Kaptanoglu Date: Wed, 5 Oct 2022 08:43:07 -0700 Subject: [PATCH] Fixed the nest conditionals in pysindy file --- pysindy/pysindy.py | 20 ++++++-------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/pysindy/pysindy.py b/pysindy/pysindy.py index b5a1e8e3d..35b699871 100644 --- a/pysindy/pysindy.py +++ b/pysindy/pysindy.py @@ -26,7 +26,6 @@ sindy_pi_flag = True except ImportError: sindy_pi_flag = False - pass from .optimizers import STLSQ from .utils import AxesArray from .utils import comprehend_axes @@ -526,11 +525,8 @@ def print(self, lhs=None, precision=3): Precision to be used when printing out model coefficients. """ eqns = self.equations(precision) - if sindy_pi_flag: - if isinstance(self.optimizer, SINDyPI): - feature_names = self.get_feature_names() - else: - feature_names = self.feature_names + if sindy_pi_flag and isinstance(self.optimizer, SINDyPI): + feature_names = self.get_feature_names() else: feature_names = self.feature_names for i, eqn in enumerate(eqns): @@ -538,16 +534,12 @@ def print(self, lhs=None, precision=3): names = "(" + feature_names[i] + ")" print(names + "[k+1] = " + eqn) elif lhs is None: - if sindy_pi_flag: - if not isinstance(self.optimizer, SINDyPI): - names = "(" + feature_names[i] + ")" - print(names + "' = " + eqn) - else: - names = feature_names[i] - print(names + " = " + eqn) - else: + if not sindy_pi_flag or not isinstance(self.optimizer, SINDyPI): names = "(" + feature_names[i] + ")" print(names + "' = " + eqn) + else: + names = feature_names[i] + print(names + " = " + eqn) else: print(lhs[i] + " = " + eqn)