Merge pull request #1796 from matthewfeickert:fix/suggest-using-extra…

…-in-error-message PiperOrigin-RevId: 615043624
tensorflow · Mar 12, 2024 · 7baa486 · 7baa486
2 parents 14a90ba + 4b480b7
commit 7baa486
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow_probability/python/__init__.py b/tensorflow_probability/python/__init__.py
@@ -83,7 +83,9 @@ def _validate_tf_environment(package):
             'installed by default when you install TensorFlow Probability. '
             'This is so that JAX-only users do not have to install TensorFlow '
             'or TF-Keras. To use TensorFlow Probability with TensorFlow, '
-            'please install the tf-keras or tf-keras-nightly package.\n\n')
+            'please install the tf-keras or tf-keras-nightly package.\n'
+            'This can be be done through installing the '
+            'tensorflow-probability[tf] extra.\n\n')
       raise