Merge branch 'master' of https://github.com/iterative/cml into runner…

…-no-special-cases
iterative · Jun 9, 2022 · cc6576d · cc6576d · github-actions · Jun 9, 2022
2 parents b811adf + f341321
commit cc6576d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/bin/cml/runner.js b/bin/cml/runner.js
@@ -501,9 +501,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',

diff --git a/bin/cml/runner.test.js b/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