fixed unit test

Author: Mohamed Zeidan

test/unit_tests/cli/test_training.py

@@ -231,7 +231,7 @@ def test_pytorch_describe_error(self, mock_hyperpod_pytorch_job):
         # Call the function and expect an exception
         result = self.runner.invoke(pytorch_describe, ["--job-name", "test-job"])
         self.assertNotEqual(result.exit_code, 0)
-        self.assertIn("Failed to describe job", result.output)
+        self.assertIn("Test error", result.output)
 
 
 @unittest.skipUnless(PYDANTIC_AVAILABLE, "Pydantic model not available")
@@ -700,4 +700,3 @@ def test_comprehensive_valid_config(self):
         self.assertEqual(config.max_retry, 3)
         self.assertEqual(len(config.volume), 1)
         self.assertEqual(config.service_account_name, "training-sa")
-