Function test_worker_set_gpu_env_for_task

tests/test_resources.py:309–328

(hq_env: HqEnv)

def test_worker_set_gpu_env_for_task(hq_env: HqEnv):
    hq_env.start_server()
    hq_env.start_worker(args=["--resource", "gpus/nvidia=[0,1]"])
    hq_env.command(
        [
            "submit",
            "--resource",
            "gpus/nvidia=2",
            "--",
            "bash",
            "-c",
            """
echo $CUDA_VISIBLE_DEVICES
""",
        ]
    )
    wait_for_job_state(hq_env, 1, "FINISHED")
    assert list(set(int(v) for v in line.split(",")) for line in read_file(default_task_output()).splitlines()) == [
        {0, 1}
    ]


@pytest.mark.parametrize(