Allow native GPU types (#1049)

Author: 0x2b3bfa0

bin/cml/runner.js

@@ -533,9 +533,9 @@ exports.builder = (yargs) =>
       },
       cloudGpu: {
         type: 'string',
-        choices: ['nogpu', 'k80', 'v100', 'tesla'],
-        coerce: (val) => (val === 'nogpu' ? undefined : val),
-        description: 'GPU type.'
+        description:
+          'GPU type. Choices: k80, v100, or native types e.g. nvidia-tesla-t4',
+        coerce: (val) => (val === 'nogpu' ? undefined : val)
       },
       cloudHddSize: {
         type: 'number',

bin/cml/runner.test.js

@@ -114,8 +114,9 @@ describe('CML e2e', () => {
                                                   cml-runner instance on the provider
                                                   i.e. tags/labels \\"key=value\\"
                                                                  [array] [default: []]
-        --cloud-gpu                               GPU type.
-                                   [string] [choices: \\"nogpu\\", \\"k80\\", \\"v100\\", \\"tesla\\"]
+        --cloud-gpu                               GPU type. Choices: k80, v100, or
+                                                  native types e.g. nvidia-tesla-t4
+                                                                              [string]
         --cloud-hdd-size                          HDD size in GB              [number]
         --cloud-ssh-private                       Custom private RSA SSH key. If not
                                                   provided an automatically generated