Resolve deprecation warnings of regex library

Signed-off-by: Emmanuel Ferdman <[email protected]>

Author: emmanuel-ferdman

pytensor/tensor/random/rewriting/jax.py

@@ -174,7 +174,7 @@ def materialize_implicit_arange_choice_without_replacement(fgraph, node):
     new_props_dict = op._props_dict().copy()
     # Signature changes from something like "(),(a),(2)->(s0, s1)" to "(a),(a),(2)->(s0, s1)"
     # I.e., we substitute the first `()` by `(a)`
-    new_props_dict["signature"] = re.sub(r"\(\)", "(a)", op.signature, 1)
+    new_props_dict["signature"] = re.sub(r"\(\)", "(a)", op.signature, count=1)
     new_op = type(op)(**new_props_dict)
     return new_op.make_node(rng, size, a_vector_param, *other_params).outputs