nit

kaito-project · May 2, 2024 · 55ea501 · 55ea501
1 parent c44955a
commit 55ea501
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/tuning/preset-tuning_test.go b/pkg/tuning/preset-tuning_test.go
@@ -276,7 +276,7 @@ func TestHandleImageDataSource(t *testing.T) {
 
 	for name, tc := range testcases {
 		t.Run(name, func(t *testing.T) {
-			initContainer, volume, volumeMount := handleImageDataSource(context.Background(), tc.workspaceObj)
+			initContainer, volume, volumeMount := handleImageDataSource(context.Background(), tc.workspaceObj.Tuning.Input.Image)
 
 			assert.Equal(t, tc.expectedInitContainerName, initContainer.Name)
 			assert.Equal(t, tc.workspaceObj.Tuning.Input.Image, initContainer.Image)