Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update error msg if TF not installed #1853

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Update error msg if TF not installed
jonas-eschle authored Nov 8, 2024
commit 193821eedf3d1822bec8bd070731b590fb4b30b1
9 changes: 3 additions & 6 deletions tensorflow_probability/python/__init__.py
Original file line number Diff line number Diff line change
@@ -38,12 +38,9 @@ def _validate_tf_environment(package):
import tensorflow as tf
except (ImportError, ModuleNotFoundError):
# Print more informative error message, then reraise.
print('\n\nFailed to import TensorFlow. Please note that TensorFlow is not '
'installed by default when you install TensorFlow Probability. This '
'is so that users can decide whether to install the GPU-enabled '
'TensorFlow package. To use TensorFlow Probability, please install '
'the most recent version of TensorFlow, by following instructions at '
'https://tensorflow.org/install.\n\n')
print('\n\nFailed to import TensorFlow, make sure it is installed. '
'For example, use `pip install tensorflow tf-keras` or follow'
'the instructions https://tensorflow.org/install.\n\n')
raise

import distutils.version