Skip to content

Commit 9937f2e

Browse files
author
Mohamed Zeidan
committed
fixed python3.9 test
1 parent b063853 commit 9937f2e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/unit_tests/cli/test_training.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,8 @@ def test_list_jobs_error(self, mock_hyperpod_pytorch_job):
205205
# Call the function and expect an exception
206206
result = self.runner.invoke(list_jobs)
207207
self.assertNotEqual(result.exit_code, 0)
208-
self.assertIn("Failed to list jobs", result.output)
208+
# Updated to match the new @handle_cli_exceptions() decorator behavior
209+
self.assertIn("Test error", result.output)
209210

210211
@patch("sagemaker.hyperpod.cli.commands.training.HyperPodPytorchJob")
211212
def test_pytorch_describe(self, mock_hyperpod_pytorch_job):

0 commit comments

Comments
 (0)