1953: added missed target_image parameter to build_docker_image method (

kubeflow#1955)
ajchili · Aug 26, 2019 · da2ecf7 · da2ecf7
1 parent 32ed27a
commit da2ecf7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/kfp/compiler/_component_builder.py b/sdk/python/kfp/compiler/_component_builder.py
@@ -439,7 +439,7 @@ def build_docker_image(staging_gcs_path, target_image, dockerfile_path, timeout=
     shutil.copyfile(dockerfile_path, dst_dockerfile_path)
 
     container_builder = ContainerBuilder(staging_gcs_path, target_image, namespace=namespace)
-    image_name_with_digest = container_builder.build(local_build_dir, dockerfile_rel_path, timeout)
+    image_name_with_digest = container_builder.build(local_build_dir, dockerfile_rel_path, target_image, timeout)
 
   logging.info('Build image complete.')
   return image_name_with_digest