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"""