mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-12-22 18:21:51 +01:00
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>
This commit is contained in:
parent
ec83d01dca
commit
b419bc4a56
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user