def __init__(self,
project_name,
experiment_name,
project_uuid,
experiment_uuid,
spec,
persistence_config=None,
outputs_refs_experiments=None,
outputs_refs_jobs=None,
experiment_group_uuid=None,
experiment_group_name=None,
original_name=None,
cloning_strategy=None,
k8s_config=None,
namespace='default',
in_cluster=False,
job_container_name=None,
job_docker_image=None,
sidecar_container_name=None,
sidecar_docker_image=None,
role_label=None,
type_label=None,
ports=None,
use_sidecar=False,
sidecar_config=None,
token_scope=None):
self.spec = spec
self.project_name = project_name
self.experiment_group_name = experiment_group_name
self.experiment_name = experiment_name
self.project_uuid = project_uuid
self.experiment_group_uuid = experiment_group_uuid
self.experiment_uuid = experiment_uuid
self.original_name = original_name
self.cloning_strategy = cloning_strategy
self.persistence_config = persistence_config
self.outputs_refs_experiments = outputs_refs_experiments
self.outputs_refs_jobs = outputs_refs_jobs
self.resource_manager = manager.ResourceManager(
namespace=namespace,
project_name=self.project_name,
experiment_group_name=self.experiment_group_name,
experiment_name=self.experiment_name,
project_uuid=self.project_uuid,
experiment_group_uuid=self.experiment_group_uuid,
experiment_uuid=experiment_uuid,
job_container_name=job_container_name,
job_docker_image=job_docker_image,
sidecar_container_name=sidecar_container_name,
sidecar_docker_image=sidecar_docker_image,
role_label=role_label,
type_label=type_label,
use_sidecar=use_sidecar,
sidecar_config=sidecar_config,
log_level=self.spec.log_level if self.spec else None,
original_name=self.original_name,
cloning_strategy=self.cloning_strategy,
declarations=self.spec.declarations if self.spec else None,
health_check_url=get_experiment_health_url(self.experiment_name))
self.token_scope = token_scope
self.ports = ports or [constants.DEFAULT_PORT]
super().__init__(k8s_config=k8s_config,
namespace=namespace,
in_cluster=in_cluster)
# Set the cluster_def
cluster_def = self.get_cluster()
self.resource_manager.set_cluster_def(cluster_def=cluster_def)
def __init__(self,
project_name,
experiment_name,
project_uuid,
experiment_uuid,
spec,
persistence_config=None,
outputs_refs_experiments=None,
outputs_refs_jobs=None,
experiment_group_uuid=None,
experiment_group_name=None,
original_name=None,
cloning_strategy=None,
k8s_config=None,
namespace='default',
in_cluster=False,
job_container_name=None,
job_docker_image=None,
sidecar_container_name=None,
sidecar_docker_image=None,
role_label=None,
type_label=None,
ports=None,
use_sidecar=False,
sidecar_config=None,
token_scope=None):
self.spec = spec
self.project_name = project_name
self.experiment_group_name = experiment_group_name
self.experiment_name = experiment_name
self.project_uuid = project_uuid
self.experiment_group_uuid = experiment_group_uuid
self.experiment_uuid = experiment_uuid
self.experiment_uuid_instance = uuid.UUID(self.experiment_uuid)
self.original_name = original_name
self.cloning_strategy = cloning_strategy
self.persistence_config = persistence_config
self.outputs_refs_experiments = outputs_refs_experiments
self.outputs_refs_jobs = outputs_refs_jobs
self.resource_manager = self.RESOURCE_MANAGER(
namespace=namespace,
project_name=self.project_name,
experiment_group_name=self.experiment_group_name,
experiment_name=self.experiment_name,
project_uuid=self.project_uuid,
experiment_group_uuid=self.experiment_group_uuid,
experiment_uuid=experiment_uuid,
job_container_name=self.get_job_container_name(job_container_name),
job_docker_image=job_docker_image,
sidecar_container_name=sidecar_container_name,
sidecar_docker_image=sidecar_docker_image,
role_label=role_label,
type_label=type_label,
use_sidecar=use_sidecar,
sidecar_config=sidecar_config,
log_level=self.spec.log_level if self.spec else None,
original_name=self.original_name,
cloning_strategy=self.cloning_strategy,
params=self.spec.params if self.spec else None,
health_check_url=get_experiment_health_url(self.experiment_name))
self.token_scope = token_scope
self.ports = self.get_ports(ports=ports)
super().__init__(k8s_config=k8s_config,
namespace=namespace,
in_cluster=in_cluster)
# Set the cluster_def
cluster_def = self.get_cluster()
self.resource_manager.set_cluster_def(cluster_def=cluster_def)
self.job_uuids = self.create_job_uuids()
def test_get_experiment_health_url(self):
self.assertEqual(
get_experiment_health_url(unique_name=self.experiment.unique_name),
'{}/_heartbeat'.format(
get_experiment_url(self.experiment.unique_name)))
