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
  
imports: import pytest
  import numpy
  import pandas
  import random
  import tempfile
  import os
  import shutil
  from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.chronos.data import TSDataset
  from pandas.testing import assert_frame_equal
  from numpy.testing import assert_array_almost_equal
label_desc: Use image node_id 2 for calling a global function with example usage: get_multi_id_ts_df() and returns: train_df