Remove call to clone state

Xilinx · Jan 11, 2025 · c42a1ed · c42a1ed
1 parent 2ce28be
commit c42a1ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(