Drop Python 3.9 and apply safe Ruff fixes

Author: Armavica

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

environment.yml

@@ -7,7 +7,7 @@ name: pytensor-dev
 channels:
   - conda-forge
 dependencies:
-  - python
+  - python>=3.10
   - compilers
   - numpy>=1.17.0
   - scipy>=0.14

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 = "[email protected]" }]
 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",
@@ -154,7 +153,7 @@ lines-after-imports = 2
 
 
 [tool.mypy]
-python_version = "3.9"
+python_version = "3.10"
 ignore_missing_imports = true
 no_implicit_optional = true
 check_untyped_defs = false

pytensor/__init__.py

@@ -76,7 +76,7 @@ def disable_log_handler(logger=pytensor_logger, handler=logging_default_handler)
 # 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/compile/builders.py

@@ -4,7 +4,7 @@
 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
@@ -304,9 +304,9 @@ def __init__(
         lop_overrides: str = "default",
         grad_overrides: str = "default",
         rop_overrides: str = "default",
-        connection_pattern: Optional[list[list[bool]]] = None,
+        connection_pattern: list[list[bool]] | None = None,
         strict: bool = False,
-        name: Optional[str] = None,
+        name: str | None = None,
         **kwargs,
     ):
         """

pytensor/compile/compilelock.py

@@ -6,7 +6,6 @@
 import os
 import threading
 from contextlib import contextmanager
-from typing import Optional, Union
 
 import filelock
 
@@ -47,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/compile/debugmode.py

@@ -13,7 +13,6 @@
 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
@@ -1354,7 +1353,7 @@ def __init__(self, maker, schedule=None):
         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:

pytensor/compile/function/pfunc.py

@@ -5,7 +5,7 @@
 
 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
@@ -105,7 +105,7 @@ def rebuild_collect_shared(
 
 
 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,
@@ -115,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],
@@ -376,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.
@@ -484,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`.
 

pytensor/compile/function/types.py

@@ -6,7 +6,7 @@
 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 @@ def __init__(
         return_none: bool,
         output_keys,
         maker: "FunctionMaker",
-        name: Optional[str] = None,
+        name: str | None = None,
     ):
         """
         Parameters
@@ -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.

pytensor/compile/mode.py

@@ -5,7 +5,7 @@
 
 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 @@ def apply(self, fgraph):
 # 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:

pytensor/compile/profiling.py

@@ -16,7 +16,7 @@
 import time
 from collections import defaultdict
 from contextlib import contextmanager
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any, Union
 
 import numpy as np
 
@@ -234,15 +234,15 @@ def reset(self):
     # Total time spent in Function.vm.__call__
     #
 
-    apply_time: Optional[dict[Union["FunctionGraph", Variable], float]] = None
+    apply_time: dict[Union["FunctionGraph", Variable], float] | None = None
 
-    apply_callcount: Optional[dict[Union["FunctionGraph", Variable], int]] = None
+    apply_callcount: dict[Union["FunctionGraph", Variable], int] | None = None
 
-    apply_cimpl: Optional[dict[Apply, bool]] = None
+    apply_cimpl: dict[Apply, bool] | None = None
     # dict from node -> bool (1 if c, 0 if py)
     #
 
-    message: Optional[str] = None
+    message: str | None = None
     # pretty string to print in summary, to identify this output
     #
 

pytensor/compile/sharedvalue.py

@@ -3,7 +3,7 @@
 import copy
 from contextlib import contextmanager
 from functools import singledispatch
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING
 
 from pytensor.graph.basic import Variable
 from pytensor.graph.utils import add_tag_trace
@@ -15,7 +15,7 @@
     from pytensor.graph.type import Type
 
 
-__SHARED_CONTEXT__: Optional[list[Variable]] = None
+__SHARED_CONTEXT__: list[Variable] | None = None
 
 
 @contextmanager
@@ -40,8 +40,8 @@ def __init__(
         value,
         strict: bool,
         allow_downcast=None,
-        container: Optional[Container] = None,
-        name: Optional[str] = None,
+        container: Container | None = None,
+        name: str | None = None,
     ):
         r"""
         Parameters
@@ -90,7 +90,7 @@ def __init__(
         if isinstance(__SHARED_CONTEXT__, list):
             __SHARED_CONTEXT__.append(self)
 
-        self._default_update: Optional[Variable] = None
+        self._default_update: Variable | None = None
 
     def get_value(self, borrow=False, return_internal_type=False):
         """
@@ -149,7 +149,7 @@ def clone(self, **kwargs):
         return cp
 
     @property
-    def default_update(self) -> Optional[Variable]:
+    def default_update(self) -> Variable | None:
         """A default update expression for this `Variable`.
 
         If this value is non-``None``, its value will be used as the `update`

pytensor/configparser.py

@@ -3,7 +3,7 @@
 import shlex
 import sys
 import warnings
-from collections.abc import Sequence
+from collections.abc import Callable, Sequence
 from configparser import (
     ConfigParser,
     InterpolationError,
@@ -13,7 +13,6 @@
 )
 from functools import wraps
 from io import StringIO
-from typing import Callable, Optional, Union
 
 from pytensor.utils import hash_from_code
 
@@ -232,10 +231,10 @@ class ConfigParam:
 
     def __init__(
         self,
-        default: Union[object, Callable[[object], object]],
+        default: object | Callable[[object], object],
         *,
-        apply: Optional[Callable[[object], object]] = None,
-        validate: Optional[Callable[[object], bool]] = None,
+        apply: Callable[[object], object] | None = None,
+        validate: Callable[[object], bool] | None = None,
         mutable: bool = True,
     ):
         """
@@ -289,7 +288,7 @@ def apply(self, value):
             return self._apply(value)
         return value
 
-    def validate(self, value) -> Optional[bool]:
+    def validate(self, value) -> bool | None:
         """Validates that a parameter values falls into a supported set or range.
 
         Raises