Add missing labels when returning TimeResponseData

joaoantoniocardoso · web-flow · commit 5cbc955b3d10 · 2023-05-20T13:47:51.000-03:00
diff --git a/control/timeresp.py b/control/timeresp.py
@@ -1111,6 +1111,8 @@ def forced_response(sys, T=None, U=0., X0=0., transpose=False,
 
     return TimeResponseData(
         tout, yout, xout, U, issiso=sys.issiso(),
+        output_labels=sys.output_index, input_labels=sys.input_index,
+        state_labels=sys.state_index,
         transpose=transpose, return_x=return_x, squeeze=squeeze)
 
 
@@ -1376,6 +1378,8 @@ def step_response(sys, T=None, X0=0., input=None, output=None, T_num=None,
 
     return TimeResponseData(
         response.time, yout, xout, uout, issiso=issiso,
+        output_labels=sys.output_index, input_labels=sys.input_index,
+        state_labels=sys.state_index,
         transpose=transpose, return_x=return_x, squeeze=squeeze)
 
 
@@ -1707,6 +1711,8 @@ def initial_response(sys, T=None, X0=0., input=0, output=None, T_num=None,
     # Store the response without an input
     return TimeResponseData(
         response.t, response.y, response.x, None, issiso=issiso,
+        output_labels=sys.output_index, input_labels=sys.input_index,
+        state_labels=sys.state_index,
         transpose=transpose, return_x=return_x, squeeze=squeeze)
 
 
@@ -1871,6 +1877,8 @@ def impulse_response(sys, T=None, X0=0., input=None, output=None, T_num=None,
 
     return TimeResponseData(
         response.time, yout, xout, uout, issiso=issiso,
+        output_labels=sys.output_index, input_labels=sys.input_index,
+        state_labels=sys.state_index,
         transpose=transpose, return_x=return_x, squeeze=squeeze)
 
 
