diff --git a/neural_compressor/jax/algorithms/dynamic.py b/neural_compressor/jax/algorithms/dynamic.py index a8b84da13d4..e9d33e419eb 100644 --- a/neural_compressor/jax/algorithms/dynamic.py +++ b/neural_compressor/jax/algorithms/dynamic.py @@ -14,6 +14,7 @@ # See the License for the specific language governing permissions and # limitations under the License. + from typing import Any, Callable, Dict, List, Optional, OrderedDict, Tuple, Union import keras