From 402dd840a1eb866d57fdf37abe2ee9130c705aef Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Mon, 30 Dec 2024 17:53:14 -0500 Subject: [PATCH] Add seed to flaky unit test. --- .../custom_modules/test_all_custom_modules.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/backend/model_manager/load/model_cache/torch_module_autocast/custom_modules/test_all_custom_modules.py b/tests/backend/model_manager/load/model_cache/torch_module_autocast/custom_modules/test_all_custom_modules.py index 9706277234..875b95da07 100644 --- a/tests/backend/model_manager/load/model_cache/torch_module_autocast/custom_modules/test_all_custom_modules.py +++ b/tests/backend/model_manager/load/model_cache/torch_module_autocast/custom_modules/test_all_custom_modules.py @@ -358,6 +358,8 @@ def patch_under_test(request: pytest.FixtureRequest) -> PatchUnderTest: def test_linear_sidecar_patches(device: str, patch_under_test: PatchUnderTest): patches, input = patch_under_test + torch.manual_seed(0) + # Build the base layer under test. layer = torch.nn.Linear(32, 64)