Skip to content

Commit 824c908

Browse files
committed
More updated in error message
1 parent ab2f855 commit 824c908

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/hyperpod/training/quota_allocation_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def _is_valid(vcpu: Optional[float], memory_in_gib: Optional[float], accelerator
249249
if instance_type is not None:
250250
#both resources and node count specified
251251
if (has_gpu_quota_allocation and node_specified):
252-
return False, f"Either node-count or a combination of accelerators, vcpu, memory-in-gib must be specified for instance-type {instance_type}"
252+
return False, f"Missing input: either node-count or a combination of accelerators, vcpu, memory-in-gib must be specified for instance-type {instance_type}"
253253
return True, ""
254254

255255

test/unit_tests/cli/test_quota_allocation_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def test_is_valid_invalid_instance_type(self):
204204
def test_is_valid_both_node_count_and_resources(self):
205205
valid, message = _is_valid(4.0, None, None, 2, "ml.g5.xlarge")
206206
assert not valid
207-
assert message == "Either node-count or a combination of accelerators, vcpu, memory-in-gib must be specified for instance-type ml.g5.xlarge"
207+
assert message == "Missing input: either node-count or a combination of accelerators, vcpu, memory-in-gib must be specified for instance-type ml.g5.xlarge"
208208

209209
def test_is_valid_both_node_count_and_limits(self):
210210
valid, message = _is_valid(None, None, None, 2, "ml.g5.xlarge")

0 commit comments

Comments
 (0)