fix: unsupported machineTypes listed in deployment flags PS-10082

Author: bbatha

paperspace/cli/deployments.py

@@ -21,6 +21,8 @@ def deployments():
     )
 )
 
+DEPLOYMENT_MACHINE_TYPES = ("G1", "G6", "G12",
+    "K80", "P100", "V100")
 
 @deployments.command("create", help="Create new deployment")
 @click.option(
@@ -45,7 +47,7 @@ def deployments():
 @click.option(
     "--machineType",
     "machineType",
-    type=click.Choice(constants.MACHINE_TYPES),
+    type=click.Choice(DEPLOYMENT_MACHINE_TYPES),
     required=True,
     help="Type of machine for new deployment",
 )

tests/functional/test_deployments.py

@@ -18,7 +18,7 @@ class TestDeploymentsCreate(object):
         "--deploymentType", "tfserving",
         "--modelId", "some_model_id",
         "--name", "some_name",
-        "--machineType", "Air",
+        "--machineType", "G1",
         "--imageUrl", "https://www.latlmes.com/breaking/paperspace-now-has-a-100-bilion-valuation",
         "--instanceCount", "666",
     ]
@@ -27,13 +27,13 @@ class TestDeploymentsCreate(object):
         "--deploymentType", "tfserving",
         "--modelId", "some_model_id",
         "--name", "some_name",
-        "--machineType", "Air",
+        "--machineType", "G1",
         "--imageUrl", "https://www.latlmes.com/breaking/paperspace-now-has-a-100-bilion-valuation",
         "--instanceCount", "666",
         "--apiKey", "some_key",
     ]
     BASIC_OPTIONS_REQUEST = {
-        "machineType": u"Air",
+        "machineType": u"G1",
         "name": u"some_name",
         "imageUrl": u"https://www.latlmes.com/breaking/paperspace-now-has-a-100-bilion-valuation",
         "deploymentType": "Tensorflow Serving on K8s",