Simple python package to shut up Tensorflow warnings and logs.
Project description
Simple python package to shut up Tensorflow warnings and logs.
How do I install this package?
As usual, just download it using pip:
pip install silence_tensorflow
Tests Coverage
Since some software handling coverages sometimes get slightly different results, here’s three of them:
How do I use it?
You only need to import the package before import Tensorflow:
from silence_tensorflow import silence_tensorflow
silence_tensorflow()
import tensorflow as tf
...
# your code
Can it be done within the import?
Sure, you can do everything with a single line by importing the submodule auto:
import silence_tensorflow.auto
import tensorflow as tf
...
# your code
How can I get pylint to ignore the unused import?
You can use the flag ‘disable=unused-import’ as such:
import silence_tensorflow.auto # pylint: disable=unused-import
import tensorflow as tf
...
# your code
How can I get pylint to ignore the unused variable?
If you import silence_tensorflow in the context of a function you will get a different warning from pyling: unused variable. You can use the flag ‘disable=unused-variable’ as such:
def func():
import silence_tensorflow.auto # pylint: disable=unused-variable
import tensorflow as tf
...
# your code
How does this work under the hood?
This package will set the KMP_AFFINITY system variable to “noverbose”, TF_CPP_MIN_LOG_LEVEL to level 3 (only errors logged).
If you need a custom value for KMP_AFFINITY you should reset it after importing the package, as follows:
import os
from silence_tensorflow import silence_tensorflow
backup = os.environ["KMP_AFFINITY"]
silence_tensorflow()
os.environ["KMP_AFFINITY"] = backup
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.