remove duplicate kwargs to docker build command

OpenMined · Jan 4, 2024 · 98cfd35 · 98cfd35
1 parent bcf6583
commit 98cfd35
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/syft/src/syft/custom_worker/builder.py b/packages/syft/src/syft/custom_worker/builder.py
@@ -106,7 +106,6 @@ def _build_image(self, tag: str, **build_opts) -> Tuple[Image, Iterable]:
             image = client.images.build(
                 tag=tag,
                 pull=True,
-                rm=True,
                 timeout=self.BUILD_MAX_WAIT,
                 **build_opts,
             )

diff --git a/packages/syft/src/syft/service/worker/utils.py b/packages/syft/src/syft/service/worker/utils.py
@@ -317,7 +317,10 @@ def create_default_image(
     """
     worker_config = DockerWorkerConfig(dockerfile=default_cpu_dockerfile)
 
-    result = image_stash.get_by_docker_config(credentials, worker_config)
+    result = image_stash.get_by_docker_config(
+        credentials=credentials,
+        config=worker_config,
+    )
 
     if result.ok() is None:
         default_syft_image = SyftWorkerImage(