From c42a1ed249e306276e5b5304ebac7114add27d88 Mon Sep 17 00:00:00 2001
From: Pablo Monteagudo Lago <pablo.monteagudolago@amd.com>
Date: Sat, 11 Jan 2025 14:21:14 +0000
Subject: [PATCH] Remove call to clone state

---
 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 1f1448f4d..8cacad219 100644
--- a/tests/brevitas/graph/test_equalization.py
+++ b/tests/brevitas/graph/test_equalization.py
@@ -454,7 +454,8 @@ 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 = generator.clone_state()
+    generator_clone = torch.Generator()
+    generator_clone.set_state(generator.get_state())
 
     # Apply rotations on the model with unfused rotations
     regions_unfused = list(