v2.0.30

phidatahq · Nov 7, 2023 · 26264d7 · 26264d7
1 parent b70cdc6
commit 26264d7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/phi/docker/resource/container.py b/phi/docker/resource/container.py
@@ -123,7 +123,17 @@ def run_container(self, docker_client: DockerApiClient) -> Optional[Any]:
         try:
             _api_client: DockerClient = docker_client.api_client
             with Progress(transient=True) as progress:
-                task = progress.add_task("Downloading Image", total=None)  # noqa: F841
+                if self.force:
+                    try:
+                        pull_image_task = progress.add_task("Downloading Image", total=None)  # noqa: F841
+                        _api_client.images.pull(self.image, platform=self.platform)
+                        progress.update(pull_image_task, completed=True)
+                    except Exception as pull_exc:
+                        logger.warning(f"Could not pull image: {self.image}: {pull_exc}")
+                if self.force:
+                    run_container_task = progress.add_task("Running Container", total=None)  # noqa: F841
+                else:
+                    run_container_task = progress.add_task("Downloading Image", total=None)  # noqa: F841
                 container_object = _api_client.containers.run(
                     name=self.name,
                     image=self.image,

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "phidata"
-version = "2.0.29"
+version = "2.0.30"
 description = "AI Toolkit for Engineers"
 requires-python = ">=3.7"
 readme = "README.md"