Skip to content

Commit 8f151ed

Browse files
committed
Fix multinode test for kserve to find gpu
Signed-off-by: Milind waykole <mwaykole@redhat.com>
1 parent 6e1e3f0 commit 8f151ed

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • tests/model_serving/model_server/kserve/multi_node

tests/model_serving/model_server/kserve/multi_node/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737

3838
@pytest.fixture(scope="session")
3939
def nvidia_gpu_nodes(nodes: list[Node]) -> list[Node]:
40-
return [node for node in nodes if "nvidia.com/gpu.present" in node.labels]
40+
return [node for node in nodes if "nvidia.com/gpu.present" in node.labels.keys()] # noqa: SIM118
4141

4242

4343
@pytest.fixture(scope="session")

0 commit comments

Comments
 (0)