SDK - Components - Fixed kfp.components.set_default_base_image (#2118)

kubeflow · Sep 16, 2019 · c4c0bb8 · c4c0bb8
1 parent 60f9da6
commit c4c0bb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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