error in running basic_example.ipynb

[boxaio]

in running the cell "Fitting a Pure Prediction RC" , an error occured:

e:\Anaconda3\envs\RCTorch\lib\site-packages\rctorch\rc.py in fit(self, y, X, burn_in, criterion, return_states, optimizer, out_weights, ODE_order, SOLVE, reparam_f, init_conditions, force, ode_coefs, train_score, ODE_criterion, q, eq_system, backprop_f, epochs, nl, n_inputs, n_outputs, random_sampling, sample_timepoints, verbose, backprop_args) 822 # append columns for the data: 823 self.extended_states = torch.cat((self.X, self.states), axis = 1) --> 824 # self.extended_states = torch.hstack((self.X, self.states)) 825 826 self.laststate = self.states[-1, :]

RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 239 but got size 56883 for tensor number 1 in the list.

it is found that the shape of self.X and the shape of self.states does not match at all