def notebook_job_pre_save(sender, **kwargs):
instance = kwargs['instance']
set_tags(instance=instance)
set_persistence(instance=instance)
set_outputs(instance=instance)
set_outputs_refs(instance=instance)
assign_code_reference(instance)
set_backend(instance=instance, default_backend=conf.get(NOTEBOOKS_BACKEND))
set_name(instance=instance, query=NotebookJob.all)
def job_pre_save(sender, **kwargs):
instance = kwargs['instance']
set_tags(instance=instance)
set_persistence(instance=instance)
set_outputs(instance=instance)
set_outputs_refs(instance=instance)
set_name(instance=instance, query=Job.all)
set_backend(instance=instance, default_backend=NATIVE_BACKEND)
assign_code_reference(instance)
def experiment_pre_save(sender, **kwargs):
instance = kwargs['instance']
# Check if declarations need to be set
if not instance.declarations and instance.specification:
instance.declarations = instance.specification.declarations
set_tags(instance=instance)
set_persistence(instance=instance)
set_outputs(instance=instance)
set_outputs_refs(instance=instance)
set_name(instance=instance, query=Experiment.all)
set_backend(instance=instance, default_backend=ExperimentBackend.NATIVE)
set_framework(instance=instance)
if not instance.specification or not instance.specification.build:
return
if instance.is_independent:
assign_code_reference(instance)
else:
instance.code_reference = instance.experiment_group.code_reference
def experiment_group_pre_save(sender, **kwargs):
instance = kwargs['instance']
# Add code reference
assign_code_reference(instance)
# Check if params need to be set
if not instance.hptuning and instance.specification:
hptuning_config = instance.specification.hptuning
hptuning = hptuning_config.to_dict()
if hptuning_config.search_algorithm == SearchAlgorithms.GRID:
hptuning['grid_search'] = hptuning.get('grid_search', {})
instance.hptuning = hptuning
set_tags(instance=instance)
set_persistence(instance=instance)
# Set type
if not instance.hptuning:
instance.group_type = GroupTypes.SELECTION
else:
instance.group_type = GroupTypes.STUDY
set_name(instance=instance, query=ExperimentGroup.all)
set_backend(instance=instance, default_backend=NATIVE_BACKEND)
def pipeline_pre_save(sender, **kwargs):
instance = kwargs['instance']
set_backend(instance=instance, default_backend=NATIVE_BACKEND)
def build_job_pre_save(sender, **kwargs):
instance = kwargs['instance']
set_tags(instance=instance)
set_backend(instance=instance, default_backend=conf.get('BUILD_BACKEND'))
assign_build_code_reference(instance)
set_name(instance=instance, query=BuildJob.all)