Python Telemanom “IndexError: too many indices for array”

Question

I am trying to test this model with some dummy inputs, my input array looks like:

X_train = np.array([[1., 2., 3.], [2., 3., 4.], [3., 4., 5.]])
X_test = np.array