Skip to content

Commit 4b480b7

Browse files
Recommend 'tensorflow-probability[tf]' extra on tf-keras import failure
* In addition to recommending users install TensorFlow dependencies piecemeal, also recommend using the 'tf' extra to install all required dependencies to use tensorflow-probability with tensorflow. * Split this off from the rest of the error message as it is already too long.
1 parent 988f023 commit 4b480b7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow_probability/python/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,9 @@ def _validate_tf_environment(package):
8383
'installed by default when you install TensorFlow Probability. '
8484
'This is so that JAX-only users do not have to install TensorFlow '
8585
'or TF-Keras. To use TensorFlow Probability with TensorFlow, '
86-
'please install the tf-keras or tf-keras-nightly package.\n\n')
86+
'please install the tf-keras or tf-keras-nightly package.\n'
87+
'This can be be done through installing the '
88+
'tensorflow-probability[tf] extra.\n\n')
8789
raise
8890

8991

0 commit comments

Comments
 (0)