Add missing import

Xilinx · Dec 6, 2023 · 61bcc70 · 61bcc70
1 parent b61b84b
commit 61bcc70
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/brevitas/graph/equalization_fixtures.py b/tests/brevitas/graph/equalization_fixtures.py
@@ -11,6 +11,7 @@
 
 from brevitas import torch_version
 from brevitas.graph.equalize import _cross_layer_equalization
+from brevitas.graph.equalize import _organize_region
 
 SEED = 123456
 ATOL = 1e-3