ndarray.ndim==0 is a scalar

python-control · Mar 30, 2021 · 5646146 · 5646146
1 parent cdba09a
commit 5646146
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/control/lti.py b/control/lti.py
@@ -665,7 +665,7 @@ def _process_frequency_response(sys, omega, out, squeeze=None):
     if squeeze is None:
         squeeze = config.defaults['control.squeeze_frequency_response']
 
-    if not hasattr(omega, '__len__'):
+    if np.asarray(omega).ndim < 1:
         # received a scalar x, squeeze down the array along last dim
         out = np.squeeze(out, axis=2)