How to fix my function to work right with timestamps?

Question

I have written a code:

This one to calculate MSE an R2 for my data:

def ComputeModel(X, y, model):
    X_train, X_test, y_train, y_test = train_test_sp