in playbooks/roles/ood-applications/files/bc_amlsdk/template/amlwrapperfunctions.py [0:0]
def runjob(filename, job_inputs):
print("Job submitted to AzureML...")
codedir = "./src/"
full_command = "python "+filename+" "+job_inputs
filename = codedir+filename
if not exists(filename):
print("Provided file does not exist:", filename)
return None, None
job_display_name = "azhop-aml-" + get_random_string(6)
job = command(
code=codedir,
command=full_command,
environment=envsetup.amlenv,
compute="gpu-cluster",
instance_count=envsetup.max_instances,
distribution=MpiDistribution(process_count_per_instance=1),
display_name=job_display_name
)
job_handler = envsetup.ml_client.create_or_update(job)
if job_handler:
print("Returning job handler to check its status. Job display name:",
job_display_name)
return job_handler, envsetup.ml_client