Skip to content

Commit

Permalink
Bump tensorflow from ==2.4.3 to ~=2.4.3 (#1015)
Browse files Browse the repository at this point in the history
### Changes

Bump tensorflow from ==2.4.3 to ~=2.4.3

### Reason for changes

Alignment with OpenVINO 2.4.2

### Related tickets

ref: 70577

### Tests

E2E 355
Manual Precommit 7
  • Loading branch information
alexsu52 committed Nov 18, 2021
1 parent d49f593 commit 870db3d
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 4 deletions.
4 changes: 3 additions & 1 deletion nncf/tensorflow/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,9 @@

import tensorflow
from pkg_resources import parse_version
if parse_version(BKC_TF_VERSION).base_version != parse_version(tensorflow.__version__).base_version:

tensorflow_version = parse_version(tensorflow.__version__).base_version
if not tensorflow_version.startswith(BKC_TF_VERSION[:-2]):
raise RuntimeError(
'NNCF only supports tensorflow=={bkc}, while current tensorflow version is {curr}'.format(
bkc=BKC_TF_VERSION,
Expand Down
2 changes: 1 addition & 1 deletion nncf/version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
__version__ = '2.0.0'
BKC_TORCH_VERSION = '1.9.1'
BKC_TORCHVISION_VERSION = '0.9.1'
BKC_TF_VERSION = '2.4.3'
BKC_TF_VERSION = '2.4.*'
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,11 +88,11 @@ def find_version(*file_paths):
version_string = "{}{}".format(sys.version_info[0], sys.version_info[1])

_extra_deps = [
"tensorflow==2.4.3",
"tensorflow~=2.4.3",
"torch>=1.5.0, <=1.9.1, !=1.8.0",
]

extra_deps = {b: a for a, b in (re.findall(r"^(([^!=<>]+)(?:[!=<>].*)?$)", x)[0] for x in _extra_deps)}
extra_deps = {b: a for a, b in (re.findall(r"^(([^~!=<>]+)(?:[~!=<>].*)?$)", x)[0] for x in _extra_deps)}

EXTRAS_REQUIRE = {
"tests": [
Expand Down

0 comments on commit 870db3d

Please sign in to comment.