tfp-nightly 0.26.0.dev20250808__py2.py3-none-any.whl → 0.26.0.dev20251103__py2.py3-none-any.whl

tensorflow_probability/python/__init__.py

@@ -47,6 +47,13 @@ def _validate_tf_environment(package):
     raise
 
   import distutils.version
+  import packaging.version
+
+  def _loose_version(version: str):
+    try:
+      return packaging.version.Version(version)
+    except packaging.version.InvalidVersion:
+      return distutils.version.LooseVersion(version)
 
   #
   # Update this whenever we need to depend on a newer TensorFlow release.
@@ -54,8 +61,9 @@ def _validate_tf_environment(package):
   required_tensorflow_version = '2.18'
 #   required_tensorflow_version = '1.15'  # Needed internally -- DisableOnExport
 
-  if (distutils.version.LooseVersion(tf.__version__) <
-      distutils.version.LooseVersion(required_tensorflow_version)):
+  if _loose_version(tf.__version__) < _loose_version(
+      required_tensorflow_version
+  ):
     raise ImportError(
         'This version of TensorFlow Probability requires TensorFlow '
         'version >= {required}; Detected an installation of version {present}. '

tensorflow_probability/python/internal/backend/jax/numpy_math.py

@@ -203,6 +203,8 @@ def _bincount(arr, weights=None, minlength=None, maxlength=None,  # pylint: disa
   """Counts number of occurences of each value in `arr`."""
   # TODO(https://github.com/google/jax/issues/5719): Use np.bincount directly?
   if not JAX_MODE:
+    if minlength is None:
+      minlength = 0
     return np.bincount(arr, weights, minlength).astype(utils.numpy_dtype(dtype))
 
   dtype = utils.numpy_dtype(dtype)
@@ -414,6 +416,8 @@ def _reduce_logsumexp(input_tensor, axis=None, keepdims=False, name=None):  # py
           or np.issubdtype(dtype, np.complexfloating)):
     # Match TF error
     raise TypeError('Input must be either real or complex')
+  if not input_tensor.shape:
+    return input_tensor
   if input_tensor.size == 0:
     # On empty arrays, mimic TF in returning `-inf` instead of failing, and
     # preserve error message if `axis` arg is incompatible with an empty array.

tensorflow_probability/python/internal/backend/numpy/numpy_math.py

@@ -203,6 +203,8 @@ def _bincount(arr, weights=None, minlength=None, maxlength=None,  # pylint: disa
   """Counts number of occurences of each value in `arr`."""
   # TODO(https://github.com/google/jax/issues/5719): Use np.bincount directly?
   if not JAX_MODE:
+    if minlength is None:
+      minlength = 0
     return np.bincount(arr, weights, minlength).astype(utils.numpy_dtype(dtype))
 
   dtype = utils.numpy_dtype(dtype)
@@ -414,6 +416,8 @@ def _reduce_logsumexp(input_tensor, axis=None, keepdims=False, name=None):  # py
           or np.issubdtype(dtype, np.complexfloating)):
     # Match TF error
     raise TypeError('Input must be either real or complex')
+  if not input_tensor.shape:
+    return input_tensor
   if input_tensor.size == 0:
     # On empty arrays, mimic TF in returning `-inf` instead of failing, and
     # preserve error message if `axis` arg is incompatible with an empty array.

tensorflow_probability/python/internal/distribution_util.py

@@ -1281,6 +1281,7 @@ def parent_frame_arguments():
 
   # Remove the *varargs, and flatten the **kwargs. Both are
   # nested lists.
+  local_vars = dict(local_vars)
   local_vars.pop(variable_arg_name, {})
   keyword_args = local_vars.pop(keyword_arg_name, {})
 

tensorflow_probability/python/version.py

@@ -34,4 +34,4 @@ __version__ = '.'.join([
 ])
 if _VERSION_SUFFIX:
   __version__ = '{}-{}'.format(__version__, _VERSION_SUFFIX)
-if __version__.endswith('dev'): __version__ += '20250808'
+if __version__.endswith('dev'): __version__ += '20251103'

tensorflow_probability/substrates/jax/internal/distribution_util.py

@@ -1281,6 +1281,7 @@ def parent_frame_arguments():
 
   # Remove the *varargs, and flatten the **kwargs. Both are
   # nested lists.
+  local_vars = dict(local_vars)
   local_vars.pop(variable_arg_name, {})
   keyword_args = local_vars.pop(keyword_arg_name, {})
 

tensorflow_probability/substrates/numpy/internal/distribution_util.py

@@ -1281,6 +1281,7 @@ def parent_frame_arguments():
 
   # Remove the *varargs, and flatten the **kwargs. Both are
   # nested lists.
+  local_vars = dict(local_vars)
   local_vars.pop(variable_arg_name, {})
   keyword_args = local_vars.pop(keyword_arg_name, {})
 

{tfp_nightly-0.26.0.dev20250808.dist-info → tfp_nightly-0.26.0.dev20251103.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tfp-nightly
-Version: 0.26.0.dev20250808
+Version: 0.26.0.dev20251103
 Summary: Probabilistic modeling and statistical inference in TensorFlow
 Home-page: http://github.com/tensorflow/probability
 Author: Google LLC

{tfp_nightly-0.26.0.dev20250808.dist-info → tfp_nightly-0.26.0.dev20251103.dist-info}/RECORD

@@ -1,6 +1,6 @@
 tensorflow_probability/__init__.py,sha256=qyOYw4eGComQ1GLNlJ0-OvaWBjQ5B0-Hjo02hlJ40VI,1222
-tensorflow_probability/python/__init__.py,sha256=tt7Wld1oUijfl-d2jQBmxfNHuuG5eHFVjvRULe4r_Zs,5690
-tensorflow_probability/python/version.py,sha256=x33i727dCQlQlMHDkMUH-Z2wdubIXBaXPP7cJMfe6pM,1476
+tensorflow_probability/python/__init__.py,sha256=qQRo06XbmO2sHYKllnDEpq9v1IhMUbO3bt9QoRyTwL8,5879
+tensorflow_probability/python/version.py,sha256=J_7mEk6VZSBiT5au2aAz1d-By2gtkVtsFMOGEQQdHiw,1476
 tensorflow_probability/python/bijectors/__init__.py,sha256=t_jgecHvPmsxGIfqfayVn8UvdhKwODhftsn7mN6SGqM,8822
 tensorflow_probability/python/bijectors/absolute_value.py,sha256=Lwfp3Px3YacTaxf3qqarsloiuMUtqugPRlVtf54wv3k,3664
 tensorflow_probability/python/bijectors/ascending.py,sha256=BeXeuA1tKDGsqP0GofGwGDwzZzYGd1I15MAsuswnc8Y,3124
@@ -362,7 +362,7 @@ tensorflow_probability/python/internal/cache_util.py,sha256=3Bg-3yyg6PrD5HbGJeql
 tensorflow_probability/python/internal/callable_util.py,sha256=e4BV15DuxzTlstGdmTd97CG_zjch32vKUf8KaZ9NO5I,2556
 tensorflow_probability/python/internal/custom_gradient.py,sha256=imYsiQra8qTY1quuSFekxjR9jzICBhsree4749rEKps,5618
 tensorflow_probability/python/internal/distribute_lib.py,sha256=RadeQBdsNtqUXYBFXE4Om6W8BJAMMXM_OXbKgxuVzfQ,13254
-tensorflow_probability/python/internal/distribution_util.py,sha256=JjXlVtd4N-SJztfeLGBLlLdmtPcWE-wNyWAQzyUuiAU,57567
+tensorflow_probability/python/internal/distribution_util.py,sha256=nNR8K_wnNUgxk4UclI0r2XUzk6Z-8BIR2z-ZBke3oq4,57599
 tensorflow_probability/python/internal/docstring_util.py,sha256=hCJRTyCoTk65vsl2kwKtgcrtT9iTVZ0LEGA-z3nznM8,1588
 tensorflow_probability/python/internal/dtype_util.py,sha256=HHUADLzQTqnGqX27SmLe5W_MF3qsnC4TbHpNbErxYRk,14758
 tensorflow_probability/python/internal/empirical_statistical_testing.py,sha256=1U78-qIbeBKuqu2C40Re0rAvqD5JBahIQT5QB2dzGbA,22679
@@ -414,7 +414,7 @@ tensorflow_probability/python/internal/backend/jax/nn.py,sha256=K9-Xy9L2dxM0cWCR
 tensorflow_probability/python/internal/backend/jax/numpy_array.py,sha256=Fos2U2Y534-EFLGu5nO73I0NPE7g5cSpVuY2XYKRhDI,16036
 tensorflow_probability/python/internal/backend/jax/numpy_keras.py,sha256=TxdcvCnfl2XqhdX5PhTOSKkE1e5y4leiNYb7Ioui9_U,856
 tensorflow_probability/python/internal/backend/jax/numpy_logging.py,sha256=ilfbgVoQImReqK7oGCJCAA2K9D4J0gIx2v0b-yfRKUQ,3496
-tensorflow_probability/python/internal/backend/jax/numpy_math.py,sha256=hk7hOXtC5hwgIwq9On15mpgi4kBACgGMBO5xcS-2fB4,31782
+tensorflow_probability/python/internal/backend/jax/numpy_math.py,sha256=V2efAtvtGMZb7QAnBFIKdn3eJMIBrcOIbgWuBbOrfaA,31881
 tensorflow_probability/python/internal/backend/jax/numpy_signal.py,sha256=ZPtGu59PxlaMBUFxHnrx1kX8EbIHAtu_YaBw6teiDEs,3442
 tensorflow_probability/python/internal/backend/jax/ops.py,sha256=tm7wDOCgGWt51y-69zBX5LR0hyDuBqHxP1qkvP3Y9HU,23480
 tensorflow_probability/python/internal/backend/jax/private.py,sha256=3bObq-w9uujbkRfLcRopOo0c8HMC1Iw5-3pDuj70HdM,4484
@@ -481,7 +481,7 @@ tensorflow_probability/python/internal/backend/numpy/nn.py,sha256=R7pSVAlteOlwYG
 tensorflow_probability/python/internal/backend/numpy/numpy_array.py,sha256=cTvK3H76ZVI2vKjSeLsXcHiGtngHg-YoFJ0tp86XrC4,16019
 tensorflow_probability/python/internal/backend/numpy/numpy_keras.py,sha256=AmaNa8HFEJZAJkOGIfrb13d2sZu_pMkPRGa7n_qHNeo,857
 tensorflow_probability/python/internal/backend/numpy/numpy_logging.py,sha256=yth3X0yznR2xg6_wIgKft2SE9T1d3zg6RGn1rbsbPBU,3497
-tensorflow_probability/python/internal/backend/numpy/numpy_math.py,sha256=ZJ-8v_z6PAqIuVel0WHdganVkWDQIdpYJpS0lHnOooQ,31761
+tensorflow_probability/python/internal/backend/numpy/numpy_math.py,sha256=Br_plGiYKGTW-UQ57QfSc18DNzVjd2cizptpWfX07mI,31860
 tensorflow_probability/python/internal/backend/numpy/numpy_signal.py,sha256=9BTjqYwR4Y4DeuWVm7RLL8AyAw9CLwA35n-EuXhVrqc,3420
 tensorflow_probability/python/internal/backend/numpy/ops.py,sha256=MeZw1cp-MNjqhec-t4Fbeq5nlsCQDiVhlEFoKeSbeIk,23456
 tensorflow_probability/python/internal/backend/numpy/private.py,sha256=CrT8Hi1WNhM5VX6DtOZpd4MozdW6rhf4Q0jYYqR4yY4,4461
@@ -959,7 +959,7 @@ tensorflow_probability/substrates/jax/internal/cache_util.py,sha256=PapeGtG-0_db
 tensorflow_probability/substrates/jax/internal/callable_util.py,sha256=vDxXtQNl4ewhALdUvgx5P_WekVqtt1zfsmrEWDB99D8,3020
 tensorflow_probability/substrates/jax/internal/custom_gradient.py,sha256=E_t8AJLMKR6CW-jwqA4SudBw2wmJ2MXBiTO05YCLYdM,6124
 tensorflow_probability/substrates/jax/internal/distribute_lib.py,sha256=RBvSxFcVcol5BrZ5Uo82TLjNT1Hqrhrqxc0CroUVFXk,13771
-tensorflow_probability/substrates/jax/internal/distribution_util.py,sha256=ODxAz8ImAAKujQW2GbndMOFa374ZgmU5e3plGwERq2Y,58106
+tensorflow_probability/substrates/jax/internal/distribution_util.py,sha256=yBVYBmMeLuxOejje42OUDrSR6sGh029c4FyS3lp3ejY,58138
 tensorflow_probability/substrates/jax/internal/dtype_util.py,sha256=0MkFifM1gXeZzI9hZS0W_4NWewAUqGfPyPTvq_paeII,15247
 tensorflow_probability/substrates/jax/internal/empirical_statistical_testing.py,sha256=C45X-XxYh2HTIFiQ3oLCOT8ehEdxVvJQKDXgjvuoMiY,23115
 tensorflow_probability/substrates/jax/internal/implementation_selection.py,sha256=avYl23pMo04siIRNZrHL5PYpPh5aC0CGiei1NA6TfuM,6404
@@ -1395,7 +1395,7 @@ tensorflow_probability/substrates/numpy/internal/cache_util.py,sha256=ABf4gQNMB1
 tensorflow_probability/substrates/numpy/internal/callable_util.py,sha256=73EYzbKYRV1LE1KNTXyWxihXDiXNWhYWbNavsNI8mHE,3024
 tensorflow_probability/substrates/numpy/internal/custom_gradient.py,sha256=MpacPtKvJ4bh_vgambxcM7n_di5_r7kqkm7g6yJJEpk,6131
 tensorflow_probability/substrates/numpy/internal/distribute_lib.py,sha256=5_deiaYNuswnQcxuNpaBFtTspzo6C9pnWDMgMpP9Gjs,13784
-tensorflow_probability/substrates/numpy/internal/distribution_util.py,sha256=oguqTLBg2vaqgs8vKvW-cm2G6JDedOBpkFPIuTmGFa8,58120
+tensorflow_probability/substrates/numpy/internal/distribution_util.py,sha256=4SbuRDExDqFCsUjtygJNDJsPeq2mRQu_qiHr7mNpyFI,58152
 tensorflow_probability/substrates/numpy/internal/dtype_util.py,sha256=fX_1AQS9YNM_ljHAUK4sUoZ-qYG-b46c0CbFcDAZHSg,15252
 tensorflow_probability/substrates/numpy/internal/empirical_statistical_testing.py,sha256=2c1_SWQuSCGLHJ8TII2UqzbIj2swIhXOKtg1GXuDm1M,23117
 tensorflow_probability/substrates/numpy/internal/implementation_selection.py,sha256=5L5nD9U4MacDFy_90VYcKQ6kIsx9wFQiODpYsc6ZwHk,6412
@@ -1535,7 +1535,7 @@ tensorflow_probability/substrates/numpy/util/seed_stream.py,sha256=Zqz5ctJoSpzHQ
 tensorflow_probability/substrates/numpy/vi/__init__.py,sha256=NNJPo76kolD21ec0xIFRROooxIKibeFh3Q9gsANzX3I,3828
 tensorflow_probability/substrates/numpy/vi/csiszar_divergence.py,sha256=804ioHtZkBBT1P65a89PABxOC-GYxYf_4ch3w7dIqKs,45484
 tensorflow_probability/substrates/numpy/vi/optimization.py,sha256=MyYaWbRzYIwVz6ka7X5nB8ryAFqvHKoRBWih-acaAbs,33690
-tfp_nightly-0.26.0.dev20250808.dist-info/METADATA,sha256=sEarXkugg3vIQ8dy2-jPgdEjzv9TDWDKgVVtRXhIxWI,13609
-tfp_nightly-0.26.0.dev20250808.dist-info/WHEEL,sha256=Ll72iyqtt6Rbxp-Q7FSafYA1LeRv98X15xcZWRsFEmY,109
-tfp_nightly-0.26.0.dev20250808.dist-info/top_level.txt,sha256=Jb6JYoMHGjOqCQ1Wdlk-IJ5vPYkPr5PsNVH8K4KjYK4,23
-tfp_nightly-0.26.0.dev20250808.dist-info/RECORD,,
+tfp_nightly-0.26.0.dev20251103.dist-info/METADATA,sha256=tDLg4kreAauMh06hHtDBQBBXn2UcOA0bZHHlXdfZIC0,13609
+tfp_nightly-0.26.0.dev20251103.dist-info/WHEEL,sha256=Ll72iyqtt6Rbxp-Q7FSafYA1LeRv98X15xcZWRsFEmY,109
+tfp_nightly-0.26.0.dev20251103.dist-info/top_level.txt,sha256=Jb6JYoMHGjOqCQ1Wdlk-IJ5vPYkPr5PsNVH8K4KjYK4,23
+tfp_nightly-0.26.0.dev20251103.dist-info/RECORD,,