reverted some changes to lose dependency on pytorch#819

ccharest93 · Apr 10, 2024 · 62c6eef · 62c6eef
1 parent 0a6c777
commit 62c6eef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/torchx/runner/api.py b/torchx/runner/api.py
@@ -377,7 +377,7 @@ def dryrun(
                     )
                 else:
                     logger.info(
-                        f"Reusing image based on `{old_img}` for role[0]={role.name}."
+                        f"Reusing original image `{old_img}` for role[0]={role.name}."
                         " Either a patch was built or no changes to workspace was detected."
                     )
 

diff --git a/torchx/workspace/docker_workspace.py b/torchx/workspace/docker_workspace.py
@@ -162,8 +162,8 @@ def build_workspace_and_update_role(
                 except StopIteration:
                     break
 
-            if len(old_imgs) == 0 or image["ID"] not in old_imgs:
-                role.image = image["ID"]
+            if len(old_imgs) == 0 or role.image not in old_imgs:
+                role.image = image.id
         finally:
             context.close()