diff --git a/lib/galaxy/tool_util/deps/requirements.py b/lib/galaxy/tool_util/deps/requirements.py index 1724001902f..e787dc4e403 100644 --- a/lib/galaxy/tool_util/deps/requirements.py +++ b/lib/galaxy/tool_util/deps/requirements.py @@ -237,7 +237,7 @@ ResourceType = Literal[ "cuda_compute_capability", "gpu_memory_min", "cuda_device_count_min", - "cuda_device_count_max" + "cuda_device_count_max", ] VALID_RESOURCE_TYPES = get_args(ResourceType) @@ -281,7 +281,7 @@ def resource_requirements_from_list(requirements) -> List[ResourceRequirement]: "cudaComputeCapability": "cuda_compute_capability", "gpuMemoryMin": "gpu_memory_min", "cudaDeviceCountMin": "cuda_device_count_min", - "cudaDeviceCountMax": "cuda_device_count_max" + "cudaDeviceCountMax": "cuda_device_count_max", } rr = [] for r in requirements: diff --git a/test/unit/tool_util/test_parsing.py b/test/unit/tool_util/test_parsing.py index b87dab14f2a..9ef3a3a2389 100644 --- a/test/unit/tool_util/test_parsing.py +++ b/test/unit/tool_util/test_parsing.py @@ -509,10 +509,19 @@ class TestYamlLoader(BaseLoaderTestCase): assert len(resource_requirements) == 6 assert resource_requirements[0].to_dict() == {"resource_type": "cores_min", "value_or_expression": 1} assert resource_requirements[1].to_dict() == {"resource_type": "cuda_version_min", "value_or_expression": 10.2} - assert resource_requirements[2].to_dict() == {"resource_type": "cuda_compute_capability", "value_or_expression": 6.1} + assert resource_requirements[2].to_dict() == { + "resource_type": "cuda_compute_capability", + "value_or_expression": 6.1, + } assert resource_requirements[3].to_dict() == {"resource_type": "gpu_memory_min", "value_or_expression": 4042} - assert resource_requirements[4].to_dict() == {"resource_type": "cuda_device_count_min", "value_or_expression": 1} - assert resource_requirements[5].to_dict() == {"resource_type": "cuda_device_count_max", "value_or_expression": 2} + assert resource_requirements[4].to_dict() == { + "resource_type": "cuda_device_count_min", + "value_or_expression": 1, + } + assert resource_requirements[5].to_dict() == { + "resource_type": "cuda_device_count_max", + "value_or_expression": 2, + } def test_outputs(self): outputs, output_collections = self._tool_source.parse_outputs(object())