From 193821eedf3d1822bec8bd070731b590fb4b30b1 Mon Sep 17 00:00:00 2001 From: Jonas Eschle Date: Fri, 8 Nov 2024 16:16:23 -0500 Subject: [PATCH] Update error msg if TF not installed --- tensorflow_probability/python/__init__.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/tensorflow_probability/python/__init__.py b/tensorflow_probability/python/__init__.py index 4255fbbb75..e74c7ae983 100644 --- a/tensorflow_probability/python/__init__.py +++ b/tensorflow_probability/python/__init__.py @@ -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