code: def _get_train_test_indexes(
      use_full_size: bool = False,
  ) -> t.Tuple[np.array, np.array]:
      """Get the indexes of the train and test sets."""
      if use_full_size:
          dataset = pd.read_csv(
              ASSETS_DIR / "just_dance_data.csv", usecols=[_TIME_COL]
          )
      else:
          dataset = pd.read_csv(
              ASSETS_DIR / "just_dance_shorted_data.csv",
              usecols=[_TIME_COL],
          )
      train_indexes = dataset[
          dataset[_TIME_COL] < _DATE_TO_SPLIT_BY
      ].index
      test_indexes = dataset[
          dataset[_TIME_COL] >= _DATE_TO_SPLIT_BY
      ].index
      return train_indexes, test_indexes
  
file_path: deepchecks/deepchecks/nlp/datasets/classification/just_dance_comment_analysis.py
imports: import pathlib
  import typing
  import warnings
  import numpy
  import pandas
  from deepchecks.nlp import TextData
  from deepchecks.utils.builtin_datasets_utils import read_and_save_data
label_desc: Use image node_id 5 for calling a global function with example usage: _get_train_test_indexes(use_full_size) and returns: train_indexes,  test_indexes