code: def get_multi_id_ts_df():
      sample_num = 100
      train_df = pd.DataFrame(
          {
              "value": np.random.randn(sample_num),
              "id": np.array(["00"] * 50 + ["01"] * 50),
              "extra feature": np.random.randn(sample_num),
          }
      )
      train_df["datetime"] = pd.date_range(
          "1/1/2019", periods=sample_num
      )
      train_df.loc[50:100, "datetime"] = pd.date_range(
          "1/1/2019", periods=50
      )
      return train_df
  
file_path: analytics-zoo/pyzoo/test/zoo/chronos/data/utils/test_roll_dataset.py
imports: import pytest
  import numpy
  import pandas
  import random
  from zoo.chronos.data import TSDataset
  from zoo.chronos.data.utils.roll_dataset import RollDataset
label_desc: Use image node_id 2 for calling a global function with example usage: get_multi_id_ts_df() and returns: train_df