From 230e9ad61a0718f6912ef2e008a5328fd298cbca Mon Sep 17 00:00:00 2001 From: Sun Choi Date: Wed, 20 Dec 2023 11:29:10 -0800 Subject: [PATCH 1/2] Fix for Falcon error from PR #587 --- optimum/habana/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index 702aea456b..5810875cc3 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(): From b70c394f261d9e9f7cf05acecd164e4df98df0e0 Mon Sep 17 00:00:00 2001 From: Sun Choi Date: Wed, 20 Dec 2023 11:43:30 -0800 Subject: [PATCH 2/2] Reformatted --- optimum/habana/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py index 5810875cc3..59b27ac984 100644 --- a/optimum/habana/utils.py +++ b/optimum/habana/utils.py @@ -340,9 +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().major == version.parse(req_version).major - ) and (get_habana_frameworks_version().minor == version.parse(req_version).minor) + 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():