Bug fix in StateSpace._remove_useless_states.

Kevin Chen · Kevin Chen · commit 01b42ce355d8 · 2011-02-08T22:18:35.000Z
Kevin K. Chen &lt;kkchen@princeton.edu&gt;
diff --git a/src/statesp.py b/src/statesp.py
@@ -157,7 +157,7 @@ def _remove_useless_states(self):
         # Remove the useless states.
         if all(useless == range(self.states)):
             # All the states were useless.
-            self.A = 0
+            self.A = zeros((1, 1))
             self.B = zeros((1, self.inputs))
             self.C = zeros((self.outputs, 1))
         else:
@@ -167,6 +167,10 @@ def _remove_useless_states(self):
             self.B = delete(self.B, useless, 0)
             self.C = delete(self.C, useless, 1)
 
+        self.states = self.A.shape[0]
+        self.inputs = self.B.shape[1]
+        self.outputs = self.C.shape[0]
+
     def copy(self):
         """Return a deep copy of the instance."""
 
