pytensor 2.19.0__tar.gz → 2.20.0__tar.gz

{pytensor-2.19.0/pytensor.egg-info → pytensor-2.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytensor
-Version: 2.19.0
+Version: 2.20.0
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license:
@@ -66,11 +66,10 @@ Classifier: Operating System :: POSIX
 Classifier: Operating System :: Unix
 Classifier: Operating System :: MacOS
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: <3.13,>=3.9
+Requires-Python: <3.13,>=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 Requires-Dist: setuptools>=48.0.0
@@ -106,8 +105,7 @@ Provides-Extra: jax
 Requires-Dist: jax; extra == "jax"
 Requires-Dist: jaxlib; extra == "jax"
 Provides-Extra: numba
-Requires-Dist: numba>=0.55; extra == "numba"
-Requires-Dist: numba-scipy>=0.3.0; extra == "numba"
+Requires-Dist: numba>=0.57; extra == "numba"
 
 .. image:: https://cdn.rawgit.com/pymc-devs/pytensor/main/doc/images/PyTensor_RGB.svg
     :height: 100px

{pytensor-2.19.0 → pytensor-2.20.0}/doc/environment.yml

@@ -3,7 +3,7 @@ channels:
   - conda-forge
   - nodefaults
 dependencies:
-  - python=3.9
+  - python=3.10
   - gcc_linux-64
   - gxx_linux-64
   - numpy

{pytensor-2.19.0 → pytensor-2.20.0}/pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "pytensor"
 dynamic = ['version']
-requires-python = ">=3.9,<3.13"
+requires-python = ">=3.10,<3.13"
 authors = [{ name = "pymc-devs", email = "pymc.devs@gmail.com" }]
 description = "Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs."
 readme = "README.rst"
@@ -30,7 +30,6 @@ classifiers = [
     "Operating System :: Unix",
     "Operating System :: MacOS",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
@@ -80,7 +79,7 @@ tests = [
 ]
 rtd = ["sphinx>=5.1.0,<6", "pygments", "pydot", "pydot2", "pydot-ng"]
 jax = ["jax", "jaxlib"]
-numba = ["numba>=0.55", "numba-scipy>=0.3.0"]
+numba = ["numba>=0.57"]
 
 [tool.setuptools.packages.find]
 include = ["pytensor*", "bin"]
@@ -154,6 +153,7 @@ lines-after-imports = 2
 
 
 [tool.mypy]
+python_version = "3.10"
 ignore_missing_imports = true
 no_implicit_optional = true
 check_untyped_defs = false

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/__init__.py

@@ -18,7 +18,6 @@ To learn more, check out:
 
 """
 
-
 __docformat__ = "restructuredtext en"
 
 # Set a default logger. It is important to do this before importing some other
@@ -77,7 +76,7 @@ from pytensor.graph.replace import clone_replace, graph_replace
 # isort: on
 
 
-def as_symbolic(x: Any, name: Optional[str] = None, **kwargs) -> Variable:
+def as_symbolic(x: Any, name: str | None = None, **kwargs) -> Variable:
     """Convert `x` into an equivalent PyTensor `Variable`.
 
     Parameters

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-03-08T17:58:24+0000",
+ "date": "2024-03-28T18:19:27+0100",
  "dirty": false,
  "error": null,
- "full-revisionid": "4ee358818543ee271b7922f5ccf988ef9338591f",
- "version": "2.19.0"
+ "full-revisionid": "ef22377d7e6ea6c74c36a8dc634ef7f1f1f8f9c6",
+ "version": "2.20.0"
 }
 '''  # END VERSION_JSON
 

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/builders.py

@@ -1,9 +1,10 @@
 """Define new Ops from existing Ops"""
+
 from collections import OrderedDict
 from collections.abc import Sequence
 from copy import copy
 from functools import partial
-from typing import Optional, cast
+from typing import cast
 
 import pytensor.tensor as pt
 from pytensor import function
@@ -91,38 +92,29 @@ def construct_nominal_fgraph(
     dict[Variable, Variable],
 ]:
     """Construct an inner-`FunctionGraph` with ordered nominal inputs."""
-    dummy_inputs = []
-    for n, inp in enumerate(inputs):
-        if (
-            not isinstance(inp, Variable)
-            or isinstance(inp, Constant)
-            or isinstance(inp, SharedVariable)
-        ):
-            raise TypeError(
-                f"Inputs and outputs must be non-Constant/shared Variable instances; got {inp}"
-            )
+    implicit_shared_inputs = []
 
-        dummy_inputs.append(inp.type())
-
-    dummy_shared_inputs = []
-    shared_inputs = []
+    dummy_inputs = [inp.type() for inp in inputs]
+    dummy_implicit_shared_inputs = []
     for var in graph_inputs(outputs, inputs):
+        if var in inputs:
+            continue
         if isinstance(var, SharedVariable):
-            # To correctly support shared variables the inner-graph should
-            # not see them; otherwise, there will be problems with
-            # gradients.
-            # That's why we collect the shared variables and replace them
-            # with dummies.
-            shared_inputs.append(var)
-            dummy_shared_inputs.append(var.type())
-        elif var not in inputs and not isinstance(var, Constant):
-            raise MissingInputError(f"OpFromGraph is missing an input: {var}")
-
-    replacements = dict(zip(inputs + shared_inputs, dummy_inputs + dummy_shared_inputs))
+            # We allow shared inputs to be added automatically to the graph
+            implicit_shared_inputs.append(var)
+            dummy_implicit_shared_inputs.append(var.type())
+        elif not isinstance(var, Constant):
+            raise MissingInputError(f"NominalGraph is missing an input: {var}")
+
+    replacements = dict(
+        zip(
+            inputs + implicit_shared_inputs, dummy_inputs + dummy_implicit_shared_inputs
+        )
+    )
 
     new = rebuild_collect_shared(
         cast(Sequence[Variable], outputs),
-        inputs=inputs + shared_inputs,
+        inputs=inputs + implicit_shared_inputs,
         replace=replacements,
         copy_inputs_over=False,
     )
@@ -132,7 +124,7 @@ def construct_nominal_fgraph(
         (clone_d, update_d, update_expr, new_shared_inputs),
     ) = new
 
-    assert len(local_inputs) == len(inputs) + len(shared_inputs)
+    assert len(local_inputs) == len(inputs) + len(implicit_shared_inputs)
     assert len(local_outputs) == len(outputs)
     assert not update_d
     assert not update_expr
@@ -154,7 +146,7 @@ def construct_nominal_fgraph(
         fgraph.clients.pop(inp, None)
         fgraph.add_input(nom_inp)
 
-    return fgraph, shared_inputs, update_d, update_expr
+    return fgraph, implicit_shared_inputs, update_d, update_expr
 
 
 class OpFromGraph(Op, HasInnerGraph):
@@ -176,8 +168,6 @@ class OpFromGraph(Op, HasInnerGraph):
         - grad() make it support DisconnectedType and the new interface
         - add support for NullType and DisconnectedType when R_op supports them
         - check how it works with updates.
-        - add test with constant as input or inside the inner graph.
-        - Add support for the GPU? Probably just need an opt to remove transfer
         - Add support to pickle this Op.
         - Add support/test with random generator
         - Add optimization to removing unused inputs/outputs
@@ -285,7 +275,7 @@ class OpFromGraph(Op, HasInnerGraph):
         #
         # For now, this converts NullType or DisconnectedType into zeros_like.
         # other types are unmodified: overrider_var -> None
-        if isinstance(grad.type, (NullType, DisconnectedType)):
+        if isinstance(grad.type, NullType | DisconnectedType):
             if hasattr(inp, "zeros_like"):
                 return inp.zeros_like(), grad
             else:
@@ -309,12 +299,14 @@ class OpFromGraph(Op, HasInnerGraph):
         self,
         inputs: list[Variable],
         outputs: list[Variable],
+        *,
         inline: bool = False,
         lop_overrides: str = "default",
         grad_overrides: str = "default",
         rop_overrides: str = "default",
-        connection_pattern: Optional[list[list[bool]]] = None,
-        name: Optional[str] = None,
+        connection_pattern: list[list[bool]] | None = None,
+        strict: bool = False,
+        name: str | None = None,
         **kwargs,
     ):
         """
@@ -398,6 +390,10 @@ class OpFromGraph(Op, HasInnerGraph):
             must be equal to number of outputs.  connection_pattern If not
             ``None``, this will be used as the connection_pattern for this
             :class:`Op`.
+        strict: bool, default False
+            If true, it raises when any variables needed to compute the inner graph
+            are not provided as explici inputs. This can only happen for graphs with
+            shared variables.
         name
             A name for debugging purposes.
         kwargs
@@ -423,6 +419,12 @@ class OpFromGraph(Op, HasInnerGraph):
             inputs, outputs
         )
 
+        if strict and self.shared_inputs:
+            raise ValueError(
+                "All variables needed to compute inner-graph must be provided as inputs under strict=True. "
+                f"The inner-graph implicitly depends on the following shared variables {self.shared_inputs}"
+            )
+
         self.kwargs = kwargs
         self.input_types = [inp.type for inp in inputs]
         self.output_types = [out.type for out in outputs]
@@ -533,7 +535,7 @@ class OpFromGraph(Op, HasInnerGraph):
             all_grads_l = list(all_grads_l)
             all_grads_ov_l = list(all_grads_ov_l)
         elif isinstance(lop_op, Variable):
-            if isinstance(lop_op.type, (DisconnectedType, NullType)):
+            if isinstance(lop_op.type, DisconnectedType | NullType):
                 all_grads_l = [inp.zeros_like() for inp in local_inputs]
                 all_grads_ov_l = [lop_op.type() for _ in range(inp_len)]
             else:
@@ -560,7 +562,7 @@ class OpFromGraph(Op, HasInnerGraph):
                     all_grads_l.append(gnext)
                     all_grads_ov_l.append(gnext_ov)
                 elif isinstance(fn_gov, Variable):
-                    if isinstance(fn_gov.type, (DisconnectedType, NullType)):
+                    if isinstance(fn_gov.type, DisconnectedType | NullType):
                         all_grads_l.append(inp.zeros_like())
                         all_grads_ov_l.append(fn_gov.type())
                     else:

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/compiledir.py

@@ -2,6 +2,7 @@
 This module contains housekeeping functions for cleaning/purging the "compiledir".
 It is used by the "pytensor-cache" CLI tool, located in the /bin folder of the repository.
 """
+
 import logging
 import os
 import pickle
@@ -54,7 +55,7 @@ def cleanup():
                                     have_npy_abi_version = True
                                 elif obj.startswith("c_compiler_str="):
                                     have_c_compiler = True
-                            elif isinstance(obj, (Op, CType)) and hasattr(
+                            elif isinstance(obj, Op | CType) and hasattr(
                                 obj, "c_code_cache_version"
                             ):
                                 v = obj.c_code_cache_version()

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/compilelock.py

@@ -2,10 +2,10 @@
 Locking mechanism to ensure no two compilations occur simultaneously
 in the same compilation directory (which can cause crashes).
 """
+
 import os
 import threading
 from contextlib import contextmanager
-from typing import Optional, Union
 
 import filelock
 
@@ -46,9 +46,9 @@ def force_unlock(lock_dir: os.PathLike):
 
 @contextmanager
 def lock_ctx(
-    lock_dir: Optional[Union[str, os.PathLike]] = None,
+    lock_dir: str | os.PathLike | None = None,
     *,
-    timeout: Optional[float] = None,
+    timeout: float | None = None,
 ):
     """Context manager that wraps around FileLock and SoftFileLock from filelock package.
 

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/debugmode.py

@@ -5,7 +5,6 @@ TODO: add support for IfElse Op, LazyLinker, etc.
 
 """
 
-
 import copy
 import gc
 import logging
@@ -14,7 +13,6 @@ from io import StringIO
 from itertools import chain
 from itertools import product as itertools_product
 from logging import Logger
-from typing import Optional
 from warnings import warn
 
 import numpy as np
@@ -681,7 +679,7 @@ def _lessbroken_deepcopy(a):
     # This logic is also in link.py
     from pytensor.link.c.type import _cdata_type
 
-    if isinstance(a, (np.ndarray, np.memmap)):
+    if isinstance(a, np.ndarray | np.memmap):
         rval = a.copy(order="K")
     elif isinstance(a, _cdata_type):
         # This is not copyable (and should be used for constant data).
@@ -1355,7 +1353,7 @@ class _Linker(LocalLinker):
         self.maker = maker
         super().__init__(scheduler=schedule)
 
-    def accept(self, fgraph, no_recycling: Optional[list] = None, profile=None):
+    def accept(self, fgraph, no_recycling: list | None = None, profile=None):
         if no_recycling is None:
             no_recycling = []
         if self.fgraph is not None and self.fgraph is not fgraph:
@@ -1891,7 +1889,7 @@ class _Linker(LocalLinker):
                         # HACK TO LOOK LIKE A REAL DESTRUCTIVE ACTION
                         # TOOK PLACE
                         if (
-                            isinstance(dr_vals[r][0], (np.ndarray, np.memmap))
+                            isinstance(dr_vals[r][0], np.ndarray | np.memmap)
                             and (dr_vals[r][0].dtype == storage_map[r][0].dtype)
                             and (dr_vals[r][0].shape == storage_map[r][0].shape)
                         ):
@@ -2021,10 +2019,10 @@ class _Maker(FunctionMaker):  # inheritance buys a few helper functions
         if outputs is None:
             return_none = True
             outputs = []
-        if not isinstance(outputs, (list, tuple)):
+        if not isinstance(outputs, list | tuple):
             unpack_single = True
             outputs = [outputs]
-        if not isinstance(inputs, (list, tuple)):
+        if not isinstance(inputs, list | tuple):
             inputs = [inputs]
 
         # Wrap them in In or Out instances if needed.
@@ -2317,10 +2315,7 @@ class DebugMode(Mode):
             raise ValueError("DebugMode has to check at least one of c and py code")
 
     def __str__(self):
-        return "DebugMode(linker={}, optimizer={})".format(
-            self.provided_linker,
-            self.provided_optimizer,
-        )
+        return f"DebugMode(linker={self.provided_linker}, optimizer={self.provided_optimizer})"
 
 
 register_mode("DEBUG_MODE", DebugMode(optimizer="fast_run"))

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/function/__init__.py

@@ -285,7 +285,7 @@ def function(
 
     if givens is None:
         givens = []
-    if not isinstance(inputs, (list, tuple)):
+    if not isinstance(inputs, list | tuple):
         raise Exception(
             "Input variables of an PyTensor function should be "
             "contained in a list, even when there is a single "
@@ -293,7 +293,7 @@ def function(
         )
 
     # compute some features of the arguments:
-    uses_tuple = any(isinstance(i, (list, tuple)) for i in inputs)
+    uses_tuple = any(isinstance(i, list | tuple) for i in inputs)
     uses_updates = bool(updates)
     uses_givens = bool(givens)
 

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/function/pfunc.py

@@ -5,7 +5,7 @@ Provide a simple user friendly API.
 
 from collections.abc import Sequence
 from copy import copy
-from typing import Optional, Union, overload
+from typing import overload
 
 from pytensor.compile.function.types import Function, UnusedInputError, orig_function
 from pytensor.compile.io import In, Out
@@ -35,8 +35,7 @@ def rebuild_collect_shared(
         list[Variable],
         list[SharedVariable],
     ],
-]:
-    ...
+]: ...
 
 
 @overload
@@ -58,8 +57,7 @@ def rebuild_collect_shared(
         list[Variable],
         list[SharedVariable],
     ],
-]:
-    ...
+]: ...
 
 
 @overload
@@ -81,8 +79,7 @@ def rebuild_collect_shared(
         list[Variable],
         list[SharedVariable],
     ],
-]:
-    ...
+]: ...
 
 
 @overload
@@ -104,12 +101,11 @@ def rebuild_collect_shared(
         list[Variable],
         list[SharedVariable],
     ],
-]:
-    ...
+]: ...
 
 
 def rebuild_collect_shared(
-    outputs: Union[Sequence[Variable], Variable, Out, Sequence[Out]],
+    outputs: Sequence[Variable] | Variable | Out | Sequence[Out],
     inputs=None,
     replace=None,
     updates=None,
@@ -119,7 +115,7 @@ def rebuild_collect_shared(
     clone_inner_graphs=False,
 ) -> tuple[
     list[Variable],
-    Union[list[Variable], Variable, Out, list[Out]],
+    list[Variable] | Variable | Out | list[Out],
     tuple[
         dict[Variable, Variable],
         dict[SharedVariable, Variable],
@@ -380,7 +376,7 @@ def pfunc(
     profile=None,
     on_unused_input=None,
     output_keys=None,
-    fgraph: Optional[FunctionGraph] = None,
+    fgraph: FunctionGraph | None = None,
 ) -> Function:
     """
     Function-constructor for graphs with shared variables.
@@ -488,7 +484,7 @@ def construct_pfunc_ins_and_outs(
     no_default_updates=False,
     rebuild_strict=True,
     allow_input_downcast=None,
-    fgraph: Optional[FunctionGraph] = None,
+    fgraph: FunctionGraph | None = None,
 ):
     """Construct inputs and outputs for `pfunc`.
 
@@ -516,7 +512,7 @@ def construct_pfunc_ins_and_outs(
     if givens is None:
         givens = []
 
-    if not isinstance(params, (list, tuple)):
+    if not isinstance(params, list | tuple):
         raise TypeError("The `params` argument must be a list or a tuple")
 
     if not isinstance(no_default_updates, bool) and not isinstance(
@@ -525,7 +521,7 @@ def construct_pfunc_ins_and_outs(
         raise TypeError("The `no_default_update` argument must be a boolean or list")
 
     if len(updates) > 0 and not all(
-        isinstance(pair, (tuple, list))
+        isinstance(pair, tuple | list)
         and len(pair) == 2
         and isinstance(pair[0], Variable)
         for pair in iter_over_pairs(updates)
@@ -579,7 +575,7 @@ def construct_pfunc_ins_and_outs(
         if outputs is None:
             out_list = []
         else:
-            if isinstance(outputs, (list, tuple)):
+            if isinstance(outputs, list | tuple):
                 out_list = list(outputs)
             else:
                 out_list = [outputs]
@@ -602,7 +598,7 @@ def construct_pfunc_ins_and_outs(
         if outputs is None:
             new_outputs = []
         else:
-            if isinstance(outputs, (list, tuple)):
+            if isinstance(outputs, list | tuple):
                 new_outputs = cloned_extended_outputs[: len(outputs)]
             else:
                 new_outputs = cloned_extended_outputs[0]

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/function/types.py

@@ -6,7 +6,7 @@ import logging
 import time
 import warnings
 from itertools import chain
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -173,7 +173,7 @@ def std_fgraph(
     input_specs: list[SymbolicInput],
     output_specs: list[SymbolicOutput],
     accept_inplace: bool = False,
-    fgraph: Optional[FunctionGraph] = None,
+    fgraph: FunctionGraph | None = None,
     features: list[type[Feature]] = [PreserveVariableAttributes],
     force_clone=False,
 ) -> tuple[FunctionGraph, list[SymbolicOutput]]:
@@ -340,7 +340,7 @@ class Function:
         return_none: bool,
         output_keys,
         maker: "FunctionMaker",
-        name: Optional[str] = None,
+        name: str | None = None,
     ):
         """
         Parameters
@@ -1310,7 +1310,7 @@ class FunctionMaker:
         elif isinstance(input, Variable):
             # r -> SymbolicInput(variable=r)
             return SymbolicInput(input)
-        elif isinstance(input, (list, tuple)):
+        elif isinstance(input, list | tuple):
             # (r, u) -> SymbolicInput(variable=r, update=u)
             if len(input) == 2:
                 return SymbolicInput(input[0], update=input[1])
@@ -1495,10 +1495,10 @@ class FunctionMaker:
         if outputs is None:
             return_none = True
             outputs = []
-        if not isinstance(outputs, (list, tuple)):
+        if not isinstance(outputs, list | tuple):
             unpack_single = True
             outputs = [outputs]
-        if not isinstance(inputs, (list, tuple)):
+        if not isinstance(inputs, list | tuple):
             inputs = [inputs]
 
         # Wrap them in In or Out instances if needed.
@@ -1693,7 +1693,7 @@ def orig_function(
     profile=None,
     on_unused_input=None,
     output_keys=None,
-    fgraph: Optional[FunctionGraph] = None,
+    fgraph: FunctionGraph | None = None,
 ) -> Function:
     """
     Return a Function that will calculate the outputs from the inputs.
@@ -1734,14 +1734,14 @@ def orig_function(
     inputs = list(map(convert_function_input, inputs))
 
     if outputs is not None:
-        if isinstance(outputs, (list, tuple)):
+        if isinstance(outputs, list | tuple):
             outputs = list(map(FunctionMaker.wrap_out, outputs))
         else:
             outputs = FunctionMaker.wrap_out(outputs)
 
     defaults = [getattr(input, "value", None) for input in inputs]
 
-    if isinstance(mode, (list, tuple)):
+    if isinstance(mode, list | tuple):
         raise ValueError("We do not support the passing of multiple modes")
 
     fn = None
@@ -1797,7 +1797,7 @@ def convert_function_input(input):
         raise TypeError(f"A Constant instance is not a legal function input: {input}")
     elif isinstance(input, Variable):
         return In(input)
-    elif isinstance(input, (list, tuple)):
+    elif isinstance(input, list | tuple):
         orig = input
         if not input:
             raise TypeError(f"Nonsensical input specification: {input}")
@@ -1806,7 +1806,7 @@ def convert_function_input(input):
             input = input[1:]
         else:
             name = None
-        if isinstance(input[0], (list, tuple)):
+        if isinstance(input[0], list | tuple):
             if len(input[0]) != 2 or len(input) != 2:
                 raise TypeError(
                     f"Invalid input syntax: {orig} (check "
@@ -1843,7 +1843,7 @@ def convert_function_input(input):
             raise TypeError(
                 f"Unknown update type: {type(update)}, expected Variable instance"
             )
-        if value is not None and isinstance(value, (Variable, SymbolicInput)):
+        if value is not None and isinstance(value, Variable | SymbolicInput):
             raise TypeError(
                 f"The value for input {variable} should not be a Variable "
                 f"or SymbolicInput instance (got: {value})"

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/io.py

@@ -3,7 +3,6 @@ Define `SymbolicInput`, `SymbolicOutput`, `In`, `Out`.
 
 """
 
-
 import logging
 
 from pytensor.link.basic import Container

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/mode.py

@@ -5,7 +5,7 @@ WRITEME
 
 import logging
 import warnings
-from typing import Literal, Optional, Union
+from typing import Literal
 
 from pytensor.compile.function.types import Supervisor
 from pytensor.configdefaults import config
@@ -260,7 +260,7 @@ optdb.register(
 # final pass just to make sure
 optdb.register("merge3", MergeOptimizer(), "fast_run", "merge", position=100)
 
-_tags: Union[tuple[str, str], tuple]
+_tags: tuple[str, str] | tuple
 
 if config.check_stack_trace in ("raise", "warn", "log"):
     _tags = ("fast_run", "fast_compile")
@@ -297,8 +297,8 @@ class Mode:
 
     def __init__(
         self,
-        linker: Optional[Union[str, Linker]] = None,
-        optimizer: Union[str, RewriteDatabaseQuery] = "default",
+        linker: str | Linker | None = None,
+        optimizer: str | RewriteDatabaseQuery = "default",
         db: RewriteDatabase = None,
     ):
         if linker is None:
@@ -565,7 +565,7 @@ def get_target_language(mode=None) -> tuple[Literal["py", "c", "numba", "jax"],
     if isinstance(linker, CLinker):
         return ("c",)
 
-    if isinstance(linker, (VMLinker, OpWiseCLinker)):
+    if isinstance(linker, VMLinker | OpWiseCLinker):
         return ("c", "py") if config.cxx else ("py",)
 
     raise Exception(f"Unsupported Linker: {linker}")

{pytensor-2.19.0 → pytensor-2.20.0}/pytensor/compile/monitormode.py

@@ -105,7 +105,7 @@ def detect_nan(fgraph, i, node, fn):
 
     for output in fn.outputs:
         if (
-            not isinstance(output[0], (np.random.RandomState, np.random.Generator))
+            not isinstance(output[0], np.random.RandomState | np.random.Generator)
             and np.isnan(output[0]).any()
         ):
             print("*** NaN detected ***")