From f3c138a208debc1a96ac5706c0118f41f7722047 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Tue, 10 Oct 2023 10:06:53 -0400 Subject: [PATCH] (minor) Fix Flake8. --- tests/backend/model_management/test_model_load_optimization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/backend/model_management/test_model_load_optimization.py b/tests/backend/model_management/test_model_load_optimization.py index 4273872069..43f007e972 100644 --- a/tests/backend/model_management/test_model_load_optimization.py +++ b/tests/backend/model_management/test_model_load_optimization.py @@ -67,4 +67,4 @@ def test_skip_torch_weight_init_restores_base_class_behavior(): _ = torch.nn.Conv1d(10, 20, 3) torch.nn.modules.conv._ConvNd.reset_parameters = saved_fn - assert called_monkey_patched_fn == True + assert called_monkey_patched_fn