Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-12-22 10:11:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

twml/util: Use a shortened condition

Browse Source 
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:
shimon-armis 2023-04-01 23:58:14 +03:00
parent ec83d01dca
commit b419bc4a56
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
twml/twml/util.py
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)