Fixed the nest conditionals in pysindy file

dynamicslab · Oct 5, 2022 · 8bfa8a0 · 8bfa8a0
1 parent a6e3903
commit 8bfa8a0
Showing 1 changed file with 6 additions and 14 deletions.
diff --git 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,28 +525,21 @@ 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):
             if self.discrete_time:
                 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)