def docker()

in manage.py [0:0]

• 29 lines of code
• 3 McCabe index (conditional complexity)

def docker(args):
    model_id, task, framework = resolve_task_framework(args)

    tag = create_docker(framework, is_gpu=args.gpu)
    gpu = ["--gpus", "all"] if args.gpu else []
    run_docker_command = [
        "docker",
        "run",
        *gpu,
        "-p",
        "8000:80",
        "-e",
        f"TASK={task}",
        "-e",
        f"MODEL_ID={model_id}",
        "-e",
        f"COMPUTE_TYPE={'gpu' if args.gpu else 'cpu'}",
        "-e",
        f"DEBUG={os.getenv('DEBUG', '0')}",
        "-v",
        "/tmp:/data",
        "-t",
        tag,
    ]

    print(" ".join(run_docker_command))
    with DockerPopen(run_docker_command) as proc:
        try:
            proc.wait()
        except KeyboardInterrupt:
            proc.terminate()