code: def setUpClass(cls):
      cls.root_dir = cls.get_base_temp_dir()
      dataset_dir = os.path.join(cls.root_dir, "musdb18hq")
      full_dataset = get_mock_dataset(dataset_dir)
      cls.train_all_samples = full_dataset["train"]
      cls.test_samples = full_dataset["test"]
      for key in cls.train_all_samples:
          if key in _VALIDATION_SET:
              cls.validation_samples[key] = cls.train_all_samples[key]
          else:
              cls.train_only_samples[key] = cls.train_all_samples[key]
  
file_path: audio/test/torchaudio_unittest/datasets/musdb_hq_test.py
imports: import os
  import torch
  from parameterized import parameterized
  from torchaudio.datasets import musdb_hq
  from torchaudio.datasets.musdb_hq import _VALIDATION_SET
  from torchaudio_unittest.common_utils import get_whitenoise, save_wav, TempDirMixin, TorchaudioTestCase
label_desc: Use image node_id 1 for calling the TestMusDB_HQ obj's underlying member method code with example usage: obj.setUpClass(cls) without return types