Skip to content

Commit cc6576d

Browse files
committed
Merge branch 'master' of https://github.com/iterative/cml into runner-no-special-cases
2 parents b811adf + f341321 commit cc6576d

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

bin/cml/runner.js

+3-3
Original file line numberDiff line numberDiff line change
@@ -501,9 +501,9 @@ exports.builder = (yargs) =>
501501
},
502502
cloudGpu: {
503503
type: 'string',
504-
choices: ['nogpu', 'k80', 'v100', 'tesla'],
505-
coerce: (val) => (val === 'nogpu' ? undefined : val),
506-
description: 'GPU type.'
504+
description:
505+
'GPU type. Choices: k80, v100, or native types e.g. nvidia-tesla-t4',
506+
coerce: (val) => (val === 'nogpu' ? undefined : val)
507507
},
508508
cloudHddSize: {
509509
type: 'number',

bin/cml/runner.test.js

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

0 commit comments

Comments
 (0)