added doc info for custom_image

yhat · Apr 26, 2017 · c30164f · c30164f
1 parent 756e6a0
commit c30164f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/yhat/api.py b/yhat/api.py
@@ -432,6 +432,8 @@ def deploy(self, name, model, session, sure=False, packages=[], patch=None, dry_
         autodetect: flag for using the requirement auto-detection feature.
             if False, you should explicitly state the packages required for
             your model, or it may not run on the server.
+        custom_image: string
+            Docker image to be used with this model (optional)
         """
         # first let's check and make sure the user actually wants to deploy
         # a new version
@@ -488,6 +490,8 @@ def deploy_tensorflow(self, name, model, session, sess, sure=False, packages=[],
         autodetect: flag for using the requirement auto-detection feature.
             if False, you should explicitly state the packages required for
             your model, or it may not run on the server.
+        custom_image: string
+            Docker image to be used with this model (optional)
         """
 
         try: