|
1 | 1 | import json |
2 | 2 | import re |
3 | 3 |
|
| 4 | +import kerastuner |
4 | 5 | import tensorflow as tf |
5 | 6 | from packaging.version import parse |
6 | 7 | from tensorflow.python.util import nest |
@@ -42,19 +43,31 @@ def to_type_key(dictionary, convert_func): |
42 | 43 |
|
43 | 44 |
|
44 | 45 | def check_tf_version() -> None: |
45 | | - if parse(tf.__version__) < parse('2.1.0'): |
| 46 | + if parse(tf.__version__) < parse('2.2.0'): |
46 | 47 | raise ImportError( |
47 | | - 'The Tensorflow package version needs to be at least v2.1.0 \n' |
| 48 | + 'The Tensorflow package version needs to be at least 2.2.0 \n' |
48 | 49 | 'for AutoKeras to run. Currently, your TensorFlow version is \n' |
49 | | - 'v{version}. Please upgrade with \n' |
50 | | - '`$ pip install --upgrade tensorflow` -> GPU version \n' |
51 | | - 'or \n' |
52 | | - '`$ pip install --upgrade tensorflow-cpu` -> CPU version. \n' |
| 50 | + '{version}. Please upgrade with \n' |
| 51 | + '`$ pip install --upgrade tensorflow`. \n' |
53 | 52 | 'You can use `pip freeze` to check afterwards that everything is ' |
54 | 53 | 'ok.'.format(version=tf.__version__) |
55 | 54 | ) |
56 | 55 |
|
57 | 56 |
|
| 57 | +def check_kt_version() -> None: |
| 58 | + if parse(kerastuner.__version__) < parse('1.0.2rc0'): |
| 59 | + raise ImportError( |
| 60 | + 'The Keras Tuner package version needs to be at least 1.0.2rc0 \n' |
| 61 | + 'for AutoKeras to run. Currently, your Keras Tuner version is \n' |
| 62 | + '{version}. Please upgrade with \n' |
| 63 | + '`$ pip install ' |
| 64 | + 'git+https://github.com/keras-team/[email protected]' |
| 65 | + '#egg=keras-tuner-1.0.2rc0`. \n' |
| 66 | + 'You can use `pip freeze` to check afterwards that everything is ' |
| 67 | + 'ok.'.format(version=kerastuner.__version__) |
| 68 | + ) |
| 69 | + |
| 70 | + |
58 | 71 | def save_json(path, obj): |
59 | 72 | obj = json.dumps(obj) |
60 | 73 | with tf.io.gfile.GFile(path, 'w') as f: |
|
0 commit comments