From 1a077caac048fbf8fcb55fc383aca772c5020c94 Mon Sep 17 00:00:00 2001 From: Pablo Monteagudo Lago Date: Sat, 11 Jan 2025 11:48:35 +0000 Subject: [PATCH] Lower PyTorch version to ensure test is run --- tests/brevitas/graph/test_equalization.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/brevitas/graph/test_equalization.py b/tests/brevitas/graph/test_equalization.py index fccd2d32d..dd7211438 100644 --- a/tests/brevitas/graph/test_equalization.py +++ b/tests/brevitas/graph/test_equalization.py @@ -12,6 +12,7 @@ import torch.nn.utils.parametrize as parametrize from torchvision import models +from brevitas import torch_version from brevitas.fx import symbolic_trace from brevitas.graph.base import ModuleInstanceRegisterParametrization from brevitas.graph.base import RotationWeightParametrization @@ -405,7 +406,7 @@ def compare_model_weights(model_fused, model_unfused, classes_to_compare=(nn.Lin assert torch.allclose(getattr(module_fused, tensor_name), getattr(module_unfused, tensor_name), atol=0.0, rtol=0.0), f"Tensor {tensor_name} does not match for module {name_module_fused}" -@requires_pt_ge('2.4') +@requires_pt_ge('2.3.1') @pytest_cases.parametrize( 'mask', itertools.product([False, True], repeat=3),