From 87861a3d80248454b465e539d0ebcb44f76d1627 Mon Sep 17 00:00:00 2001 From: Kien Dang Date: Thu, 16 May 2024 19:28:41 +0800 Subject: [PATCH] Fix worker pool integration tests set registry_uid --- tests/integration/container_workload/pool_image_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/container_workload/pool_image_test.py b/tests/integration/container_workload/pool_image_test.py index c01532bcf49..d4be44bbf01 100644 --- a/tests/integration/container_workload/pool_image_test.py +++ b/tests/integration/container_workload/pool_image_test.py @@ -140,7 +140,7 @@ def test_pool_launch( docker_build_result = domain_client.api.services.worker_image.build( image_uid=worker_image.id, tag=docker_tag, - pull=False, + registry_uid=external_registry_uid, ) assert isinstance(docker_build_result, SyftSuccess) @@ -236,7 +236,7 @@ def test_pool_image_creation_job_requests( tag=docker_tag, config=worker_config, reason="I want to do some more cool data science with PySyft", - pull_image=False, + registry_uid=external_registry_uid, ) assert isinstance(request, Request) assert len(request.changes) == 2