code: def __init__(self, training_script, *args, **kwargs):
      self._training_script = training_script
      super(BaseElasticTests, self).__init__(*args, **kwargs)
  
imports: import contextlib
  import json
  import os
  import sys
  from typing import Callable
  import mock
  import pytest
  from horovod.runner.common.util import config_parser
  from horovod.runner.elastic import constants
  from horovod.runner.launch import parse_args, _run_elastic
  from common import override_args, temppath
label_desc: Use image node_id 1 to create a new BaseElasticTests object  with example: obj = BaseElasticTests(training_script)