INNER CODE UNIT · Python

create_training_job

GACWR/OpenUBA · core/operator/main.py:166

def create_training_job(spec, name, meta, status, **kwargs):
    runtime = spec.get('runtime', 'python-base')
    config_path = spec.get('configPath')
    output_path = spec.get('outputPath')
    model_ref = spec.get('modelRef')

    # Resolve runtime
    image_tag = "base"
    if runtime in ["sklearn", "pytorch", "tensorflow", "networkx"]:
        image_tag = runtime
    image = f"openuba-model-runner:{image_tag}"

    job_name = f"train-job-{name}"

    # Env vars for training
    db_url = os.getenv("DATABASE_URL", "postgresql://gacwr:gacwr@postgres.openuba.svc:5432/openuba")
    run_id = spec.get('runId', '')
    execution_id = spec.get('executionId', '')

View source record →