Compare commits

...

2 Commits

Author SHA1 Message Date
Shimon Turjeman
d1df335c9a
Merge b419bc4a56 into fb54d8b549 2023-05-22 17:38:11 -05:00
shimon-armis
b419bc4a56 twml/util: Use a shortened condition
This commit enhances an assertion on log level
by using a shortened version for range comparison:
x <= y <= z
Instead of
y => x and y <= z

Signed-off-by: shimon-armis <shimon.turjeman@armis.com>
2023-04-01 23:58:14 +03:00

View File

@ -481,7 +481,7 @@ def set_tensorflow_log_level(log_level):
Note that tf.Print output are INFO logs, so setting log_level above 0 would hide Note that tf.Print output are INFO logs, so setting log_level above 0 would hide
output from tf.Print. output from tf.Print.
""" """
assert isinstance(log_level, int) and log_level >= 0 and log_level <= 3 assert isinstance(log_level, int) and 0 <= log_level <= 3
os.environ['TF_CPP_MIN_LOG_LEVEL'] = str(log_level) os.environ['TF_CPP_MIN_LOG_LEVEL'] = str(log_level)