Allow user specify existing image (#501)

Author: jinchihe

kubeflow/fairing/__init__.py

@@ -8,4 +8,4 @@
 
 name = "fairing"
 
-__version__ = "0.7.2"
+__version__ = "0.7.3"

kubeflow/fairing/builders/base_builder.py

@@ -38,11 +38,13 @@ def __init__(self,
         self.image_tag = None
         self.docker_client = None
 
-    def generate_pod_spec(self):
+    def generate_pod_spec(self, image=None):
+        if not image:
+            image = self.image_tag
         return client.V1PodSpec(
             containers=[client.V1Container(
                 name='model',
-                image=self.image_tag,
+                image=image,
                 command=self.preprocessor.get_command(),
                 security_context=client.V1SecurityContext(
                     run_as_user=0,

kubeflow/fairing/builders/builder.py

@@ -10,7 +10,7 @@ def build(self):
         raise NotImplementedError('BuilderInterface.build')
 
     @abc.abstractmethod
-    def generate_pod_spec(self):
+    def generate_pod_spec(self, image=None):
         """This method should return a V1PodSpec with the correct image set.
             This is also where the builder should set the environment variables
             and volume/volumeMounts that it may need to work

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='kubeflow-fairing',
-    version='0.7.2',
+    version='0.7.3',
     author="Kubeflow Authors",
     author_email='[email protected]',
     license="Apache License Version 2.0",