The code below gives a ValueError: Input variables with an inconsistent number of samples were found: [8082, 5572]. Fix it! from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test = train_test_split(tfidf, cats, test_size=0.3, random_state=42)

def train_test_split(X, y, test_size=0.3, random_state=42):
  """
  Input:
    X: A pandas dataframe.
    y: A pandas series. Corresponds to the labels.
    test_size: Test size as a fraction of the total dataset.
  """
  # YOUR CODE HERE
  raise NotImplementedError()
  return X_train, X_test, y_train, y_test