Allow native GPU types

iterative · Jun 6, 2022 · 195351c · 195351c · github-actions · Jun 6, 2022
1 parent 1559f48
commit 195351c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bin/cml/runner.js b/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]. Also supports native types like e.g. nvidia-tesla-t4',
+        coerce: (val) => (val === 'nogpu' ? undefined : val)
       },
       cloudHddSize: {
         type: 'number',