[wanda] fix a bug of local image resolving

otherwise it is breaking backwards compatibility
ray-project · Aug 17, 2023 · e1bb6af · e1bb6af
1 parent 572843b
commit e1bb6af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/wanda/resolve_image.go b/wanda/resolve_image.go
@@ -24,8 +24,9 @@ func resolveLocalImage(name, ref string) (*imageSource, error) {
 	}
 
 	return &imageSource{
-		name: name,
-		id:   id.String(),
+		name:  name,
+		id:    id.String(),
+		local: ref,
 	}, nil
 }
 

diff --git a/wanda/resolve_image_test.go b/wanda/resolve_image_test.go
@@ -48,6 +48,9 @@ func TestResolveLocalImage(t *testing.T) {
 	if want := imageID.String(); want != src.id {
 		t.Errorf("got image id %q, want %q", src.id, want)
 	}
+	if src.local != tagStr {
+		t.Errorf("got image local %q, want %q", src.local, tagStr)
+	}
 
 	dockerCmd := newDockerCmd("")
 	if err := dockerCmd.run("image", "rm", tagStr); err != nil {