def __init__(self, args): namespace, name, env = test_runner.parse_runtime_params(args) self.app_dir = args.app_dir self.env = env self.namespace = namespace self.tfjob_version = args.tfjob_version self.params = args.params super(EstimatorRunconfigTests, self).__init__(class_name="EstimatorRunconfigTests", name=name)
def __init__(self, args): namespace, name, env = test_runner.parse_runtime_params(args) self.app_dir = args.app_dir self.env = env self.namespace = namespace self.tfjob_version = args.tfjob_version self.params = args.params super(ReplicaRestartPolicyTests, self).__init__(class_name="ReplicaRestartPolicyTests", name=name)
def __init__(self, args): namespace, name, env = test_runner.parse_runtime_params(args) self.app_dir = args.app_dir self.env = env self.namespace = namespace self.tfjob_version = args.tfjob_version self.params = args.params super(DistributedTrainingJobTests, self).__init__(class_name="DistributedTrainingJobTests", name=name)
def __init__(self, args): namespace, name, env = test_runner.parse_runtime_params(args) self.app_dir = args.app_dir self.cluster_name = args.cluster self.env = env self.namespace = namespace self.tfjob_version = args.tfjob_version self.params = args.params self.failure = None logging.info("env = %s", str(self.env)) logging.info("params = %s", str(self.params)) super(PodNamesValidationTest, self).__init__(class_name="PodNamesValidationTest", name=name)
def __init__(self, args): namespace, name, env = test_runner.parse_runtime_params(args) self.app_dir = args.app_dir if not self.app_dir: self.app_dir = os.path.join(os.path.dirname(__file__), "..", "ks_app") self.app_dir = os.path.abspath(self.app_dir) logging.info("--app_dir not set defaulting to: %s", self.app_dir) self.env = env self.namespace = namespace self.params = args.params self.ks_cmd = ks_util.get_ksonnet_cmd(self.app_dir) super(TFJobTest, self).__init__(class_name="TFJobTest", name=name)