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)
