Fix worker pool integration tests

set registry_uid
OpenMined · May 16, 2024 · 87861a3 · 87861a3
1 parent 16bcf91
commit 87861a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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