DMD with control cannot reconstruct data when provided with control

[lcstrang]

Hi,

Beginner PyDMD user, so apologies if this is user-error.

The problem
DMD with control cannot reconstruct data if provided with a control input, even if the input is identical to the control used to fit.

Example
Code snippet:
dmdc = DMDc(svd_rank=0, opt=True).fit(X=train, I=control) dmdc.reconstructed_data(control).real

Expected behavior
Output of a reconstructed data matrix

Output
in DMDc.reconstructed_data(self, control_input) 274 arr = A.dot(data[i]) + self._B.dot(u) 275 if arr.shape != expected_shape: --> 276 raise ValueError( 277 f"Invalid shape: expected {expected_shape}, got {arr.shape}" 278 ) 279 data.append(arr) 281 data = np.array(data).T
ValueError: Invalid shape: expected (20592,), got (20592, 20592)
20592 matches the expected dimension of one time-step of the training data.

Additional context
The code performs as expected if I do not provide a control matrix