Skip to content

Commit c6b408c

Browse files
pintaoz-awspintaoz
andauthored
Add cleanup in test_invalid_no_node_count_or_quota_parameter() (#356)
Co-authored-by: pintaoz <[email protected]>
1 parent 7095543 commit c6b408c

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

test/integration_tests/training/cli/test_gpu_quota_allocation.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,15 @@ def test_invalid_no_node_count_or_quota_parameter(self, test_job_name):
260260
)
261261
assert result.returncode == 0
262262

263+
delete_cmd = [
264+
"hyp", "delete", "hyp-pytorch-job",
265+
"--job-name", test_job_name,
266+
"--namespace", NAMESPACE
267+
]
268+
result = execute_command(delete_cmd)
269+
assert result.returncode == 0
270+
logger.info(f"Successfully deleted job: {test_job_name}")
271+
263272
def test_invalid_instance_type_parameter(self, test_job_name):
264273
"""Test case where invalid instance type parameter is provided"""
265274

0 commit comments

Comments
 (0)