code: def __init__(self, data_dir):
      self.train_file = data_file(data_dir, "train_small")
      self.test_file = data_file(data_dir, "test_small")
      self.cd_file = data_file(data_dir, "train.cd")
  
imports: import os
  import pytest
  import re
  import yatest.common
  from lib import data_file, get_limited_precision_dsv_diff_tool
label_desc: Use image node_id 1 to create a new Dataset object from inherited base classes: object with example: obj = Dataset(data_dir)