fix

Author: echarlaix

tests/neural_compressor/test_optimization.py

@@ -18,6 +18,7 @@
 import os
 import tempfile
 
+import unittest
 import evaluate
 import numpy as np
 import torch