Repository navigation
Commit fcdc7da
authored
fix: add training_plan_arn to compute MagicMock in serverful tests (#6270)
The training_plan_arn passthrough fix causes Pydantic validation errors
in tests that use MagicMock for the compute object without explicitly
setting training_plan_arn. MagicMock returns a MagicMock object instead
of None/string, which Pydantic rejects.
Add training_plan_arn=None to the compute MagicMock in
test_base_trainer_serverful.py and test_serverful_recipe_validation.py.1 parent 8d924cb commit fcdc7da
2 files changed
Lines changed: 2 additions & 0 deletions
File tree
- sagemaker-train/tests/unit/train
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
31 | 31 | | |
32 | 32 | | |
33 | 33 | | |
| 34 | + | |
34 | 35 | | |
35 | 36 | | |
36 | 37 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
41 | 41 | | |
42 | 42 | | |
43 | 43 | | |
| 44 | + | |
44 | 45 | | |
45 | 46 | | |
46 | 47 | | |
| |||
0 commit comments