From 935cdd8d5c50cbddbd474249179b03a4cb1432ed Mon Sep 17 00:00:00 2001 From: pintaoz Date: Fri, 19 Dec 2025 11:26:49 -0800 Subject: [PATCH] Add cleanup in test_invalid_no_node_count_or_quota_parameter() --- .../training/cli/test_gpu_quota_allocation.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/test/integration_tests/training/cli/test_gpu_quota_allocation.py b/test/integration_tests/training/cli/test_gpu_quota_allocation.py index a2a5e912..c4ea1480 100644 --- a/test/integration_tests/training/cli/test_gpu_quota_allocation.py +++ b/test/integration_tests/training/cli/test_gpu_quota_allocation.py @@ -260,6 +260,15 @@ def test_invalid_no_node_count_or_quota_parameter(self, test_job_name): ) assert result.returncode == 0 + delete_cmd = [ + "hyp", "delete", "hyp-pytorch-job", + "--job-name", test_job_name, + "--namespace", NAMESPACE + ] + result = execute_command(delete_cmd) + assert result.returncode == 0 + logger.info(f"Successfully deleted job: {test_job_name}") + def test_invalid_instance_type_parameter(self, test_job_name): """Test case where invalid instance type parameter is provided"""