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(InvalidTfJobTests, self).__init__(class_name="InvalidTfJobTests",
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__), "..",
"training/GCS")
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
super(TFJobTest, self).__init__(class_name="TFJobTest", name=name)
