From 98bcf394be140bbf1f2b1e22c2d27af6ec1ac9b7 Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Fri, 13 Sep 2019 16:51:05 -0700 Subject: [PATCH] SDK - Components - Fixed kfp.components.set_default_base_image --- sdk/python/kfp/components/_python_op.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdk/python/kfp/components/_python_op.py b/sdk/python/kfp/components/_python_op.py index 2070ad47c2d..bc9ebf0951f 100644 --- a/sdk/python/kfp/components/_python_op.py +++ b/sdk/python/kfp/components/_python_op.py @@ -53,6 +53,7 @@ def set_default_base_image(image_or_factory: Union[str, Callable[[], str]]): '''set_default_base_image sets the name of the container image that will be used for component creation when base_image is not specified. Alternatively, the base image can also be set to a factory function that will be returning the image. ''' + global _default_base_image _default_base_image = image_or_factory