Allow only tensorflow deployments

BartoszCki · BartoszCki · commit 9a72f1fcfef9 · 2019-05-28T19:07:14.000+02:00
diff --git a/Pipfile b/Pipfile
@@ -16,7 +16,7 @@ terminaltables = "*"
 click-didyoumean = "*"
 click-help-colors = "*"
 colorama = "*"
-
+click-completion = "*"
 
 [dev-packages]
 twine = "*"
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/paperspace/cli/deployments.py b/paperspace/cli/deployments.py
@@ -17,7 +17,7 @@ def deployments():
 DEPLOYMENT_TYPES_MAP = collections.OrderedDict(
     (
         ("TFSERVING", "Tensorflow Serving on K8s"),
-        ("GRADIENT", "Gradient Jobs"),
+        # ("GRADIENT", "Gradient Jobs"),
     )
 )
 
@@ -31,7 +31,7 @@ def deployments():
     "deploymentType",
     type=ChoiceType(DEPLOYMENT_TYPES_MAP, case_sensitive=False),
     required=True,
-    help="Model deployment type",
+    help="Model deployment type. Only TensorFlow models can currently be deployed",
 )
 @click.option(
     "--modelId",

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def deployments():`
`17`	`17`	`DEPLOYMENT_TYPES_MAP = collections.OrderedDict(`
`18`	`18`	`(`
`19`	`19`	`("TFSERVING", "Tensorflow Serving on K8s"),`
`20`		`- ("GRADIENT", "Gradient Jobs"),`
	`20`	`+ # ("GRADIENT", "Gradient Jobs"),`
`21`	`21`	`)`
`22`	`22`	`)`
`23`	`23`
`@@ -31,7 +31,7 @@ def deployments():`
`31`	`31`	`"deploymentType",`
`32`	`32`	`type=ChoiceType(DEPLOYMENT_TYPES_MAP, case_sensitive=False),`
`33`	`33`	`required=True,`
`34`		`- help="Model deployment type",`
	`34`	`+ help="Model deployment type. Only TensorFlow models can currently be deployed",`
`35`	`35`	`)`
`36`	`36`	`@click.option(`
`37`	`37`	`"--modelId",`