Commit f75c4fbb authored by mohamedadelnaguib's avatar mohamedadelnaguib
Browse files

Fixing an assertion error happen when the number of inout equals to and the axs isn't None

parent f512a914
......@@ -272,6 +272,8 @@ class TransferFunction(CacheableObject, ParametricObject):
fig.set_constrained_layout(True)
ax = fig.subplots(2 * num_output, num_input, sharex=True, squeeze=False)
else:
if num_input == 1:
ax = ax.reshape((2*num_output, 1))
assert isinstance(ax, np.ndarray)
assert ax.shape == (2 * num_output, num_input), \
f'ax.shape={ax.shape} should be ({2 * num_output}, {num_input})'
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment