diff --git a/docs/contents/workers/yaml.md b/docs/contents/workers/yaml.md index b4ce1bc97fdf2f95ce22bc18701bd64fefacb1c9..b7018c7c0573e3501e214f30065c5ce80c6eb573 100644 --- a/docs/contents/workers/yaml.md +++ b/docs/contents/workers/yaml.md @@ -80,7 +80,16 @@ include: : This worker does not support GPUs. It may run on a host that has a GPU, but it will ignore it. `model_usage` -: Boolean. Whether or not this worker requires a model version to run. Defaults to `false`. +: Whether or not this worker requires a model version to run. Defaults to `disabled`. May take one of the following values: + + `required` + : This worker requires a model version, and will only be run on processes with a model. + + `supported` + : This worker supports a model version, but may run on any processes, including those without model. + + `disabled` + : This worker does not support model version. It may run on a process that has a model, but it will ignore it. `docker` : Regroups Docker-related configuration attributes: diff --git a/tests/conftest.py b/tests/conftest.py index 92dc737891a863bab21ab71ea1f0a421118613ad..4155fbe1be7032087b855e6da7f823f7bab2b077 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -139,7 +139,7 @@ def _mock_worker_run_api(responses): "docker_image_name": None, "state": "created", "gpu_usage": "disabled", - "model_usage": False, + "model_usage": "disabled", "worker": { "id": "deadbeef-1234-5678-1234-worker", "name": "Fake worker", @@ -206,7 +206,7 @@ def _mock_worker_run_no_revision_api(responses): "docker_image_name": None, "state": "created", "gpu_usage": "disabled", - "model_usage": False, + "model_usage": "disabled", "worker": { "id": "deadbeef-1234-5678-1234-worker", "name": "Fake worker",