blob: 669028f1d674b55da6d1e43cf8c4ef4f636b27ef (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
"""OpenNMT-tf version."""
__version__ = "2.25.0"
INCLUSIVE_MIN_TF_VERSION = "2.4.0"
EXCLUSIVE_MAX_TF_VERSION = "2.9.0"
def _check_tf_version():
import warnings
from distutils.version import LooseVersion
import tensorflow as tf
if (
LooseVersion(INCLUSIVE_MIN_TF_VERSION)
<= LooseVersion(tf.__version__)
< LooseVersion(EXCLUSIVE_MAX_TF_VERSION)
):
return
warnings.warn(
"OpenNMT-tf supports TensorFlow versions %s (included) to %s (excluded), "
"but you have TensorFlow %s installed. Some features might not work properly."
% (INCLUSIVE_MIN_TF_VERSION, EXCLUSIVE_MAX_TF_VERSION, tf.__version__),
UserWarning,
)
|