diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index 702aea456b..59b27ac984 100644 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -340,7 +340,9 @@ def check_habana_frameworks_version(req_version): """ Checks if the installed version of `habana_frameworks` is equal to `req_version`. """ - return get_habana_frameworks_version() == version.parse(req_version) + return (get_habana_frameworks_version().major == version.parse(req_version).major) and ( + get_habana_frameworks_version().minor == version.parse(req_version).minor + ) def get_device_name():