From 962b49bbdeca2b764546440eebe563baa3bf1b1f Mon Sep 17 00:00:00 2001 From: xinhe3 Date: Tue, 26 Nov 2024 03:50:55 +0200 Subject: [PATCH] make style Signed-off-by: xinhe3 --- examples/text-generation/utils.py | 2 +- optimum/habana/utils.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/text-generation/utils.py b/examples/text-generation/utils.py index c6a5bbe7cb..94069b9f0e 100644 --- a/examples/text-generation/utils.py +++ b/examples/text-generation/utils.py @@ -38,9 +38,9 @@ ) from optimum.habana.utils import ( check_habana_frameworks_version, + check_neural_compressor_min_version, check_optimum_habana_min_version, get_habana_frameworks_version, - check_neural_compressor_min_version, set_seed, ) diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index ff0d96cb79..01cd26f93d 100755 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -389,6 +389,7 @@ def check_neural_compressor_min_version(req_version): Checks if the installed version of `neural_compressor` is larger than or equal to `req_version`. """ import neural_compressor + return version.Version(neural_compressor.__version__) >= version.Version(req_version)