Compliancy with tf-nightly (#9570)

* Compliancy with tf-nightly * Add more version + restore min version check

Compliancy with tf-nightly (#9570)
* Compliancy with tf-nightly * Add more version + restore min version check
14d677ca · Julien Plu · GitHub · 46ed56cf · 14d677ca
Unverified Commit 14d677ca authored Jan 14, 2021 by Julien Plu Committed by GitHub Jan 14, 2021
Show whitespace changes
Inline Side-by-side

Showing with 14 additions and 2 deletions

src/transformers/file_utils.py src/transformers/file_utils.py +14 -2

No files found.
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -88,6 +88,18 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
        except importlib_metadata.PackageNotFoundError:
            try:
                _tf_version = importlib_metadata.version("tensorflow-cpu")
+            except importlib_metadata.PackageNotFoundError:
+                try:
+                    _tf_version = importlib_metadata.version("tensorflow-gpu")
+                except importlib_metadata.PackageNotFoundError:
+                    try:
+                        _tf_version = importlib_metadata.version("tf-nightly")
+                    except importlib_metadata.PackageNotFoundError:
+                        try:
+                            _tf_version = importlib_metadata.version("tf-nightly-cpu")
+                        except importlib_metadata.PackageNotFoundError:
+                            try:
+                                _tf_version = importlib_metadata.version("tf-nightly-gpu")
                            except importlib_metadata.PackageNotFoundError:
                                _tf_version = None
                                _tf_available = False