We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9a8b631 commit 17eeefaCopy full SHA for 17eeefa
1 file changed
tests/model_registry/conftest.py
@@ -320,15 +320,12 @@ def model_registry_deployment_containers(model_registry_namespace: str) -> list[
320
321
@pytest.fixture(scope="class")
322
def model_registry_pod(admin_client: DynamicClient, model_registry_namespace: str) -> Pod:
323
- mr_pod = list(
324
- Pod.get(
325
- dyn_client=admin_client,
326
- namespace=model_registry_namespace,
327
- label_selector=MODEL_REGISTRY_POD_FILTER,
328
- )
329
330
- assert len(mr_pod) == 1
331
- return mr_pod[0]
+ return wait_for_pods_by_labels(
+ admin_client=admin_client,
+ namespace=model_registry_namespace,
+ label_selector=MODEL_REGISTRY_POD_FILTER,
+ expected_num_pods=1,
+ )[0]
332
333
334
0 commit comments