pytensor 2.30.0__tar.gz → 2.30.1__tar.gz

{pytensor-2.30.0/pytensor.egg-info → pytensor-2.30.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 2.30.0
+Version: 2.30.1
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license:

{pytensor-2.30.0 → pytensor-2.30.1}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-03-28T02:16:52+0800",
+ "date": "2025-03-30T18:28:57+0800",
  "dirty": false,
  "error": null,
- "full-revisionid": "0b56ed9c6514e4ba641cb07fd7bbca629dd797c3",
- "version": "2.30.0"
+ "full-revisionid": "3af923ba1593320cf0a5bfb29a04263406652731",
+ "version": "2.30.1"
 }
 '''  # END VERSION_JSON
 

{pytensor-2.30.0 → pytensor-2.30.1}/pytensor/link/jax/dispatch/random.py

@@ -409,12 +409,14 @@ def jax_sample_fn_multinomial(op, node):
         sampling_rng = jax.random.split(rng_key, binom_p.shape[0])
 
         def _binomial_sample_fn(carry, p_rng):
-            s, rho = carry
+            remaining_n, remaining_p = carry
             p, rng = p_rng
-            samples = jax.random.binomial(rng, s, p / rho)
-            s = s - samples
-            rho = rho - p
-            return ((s, rho), samples)
+            samples = jnp.where(
+                p == 0, 0, jax.random.binomial(rng, remaining_n, p / remaining_p)
+            )
+            remaining_n -= samples
+            remaining_p -= p
+            return ((remaining_n, remaining_p), samples)
 
         (remain, _), samples = jax.lax.scan(
             _binomial_sample_fn,

{pytensor-2.30.0 → pytensor-2.30.1/pytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 2.30.0
+Version: 2.30.1
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license: