From 2ce28bec1515c40bd141309b51d6a6397904d04f Mon Sep 17 00:00:00 2001 From: Pablo Monteagudo Lago Date: Sat, 11 Jan 2025 12:22:43 +0000 Subject: [PATCH] Revert change to generator --- tests/brevitas/graph/test_equalization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/brevitas/graph/test_equalization.py b/tests/brevitas/graph/test_equalization.py index dd7211438..1f1448f4d 100644 --- a/tests/brevitas/graph/test_equalization.py +++ b/tests/brevitas/graph/test_equalization.py @@ -454,7 +454,7 @@ def test_apply_rotate( generator = torch.Generator() generator.manual_seed(SEED) # Clone generator to make sure we can use the same rotation matrices - generator_clone = copy.deepcopy(generator) + generator_clone = generator.clone_state() # Apply rotations on the model with unfused rotations regions_unfused = list(