Upgrade to ruff 0.2.0 and fix RUF017

* --show-source -> --output-format=full
* renaming of some config options
* removing --line-length because it is already in the pyproject file
* taking care of some list quadratic summations

Author: Armavica

.pre-commit-config.yaml

@@ -20,12 +20,11 @@ repos:
           )$
       - id: check-merge-conflict
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.14
+    rev: v0.2.0
     hooks:
       - id: ruff
-        args: ["--fix", "--show-source"]
+        args: ["--fix", "--output-format=full"]
       - id: ruff-format
-        args: ["--line-length=88"]
   - repo: https://github.com/pre-commit/mirrors-mypy
     rev: v1.8.0
     hooks:

pyproject.toml

@@ -121,20 +121,16 @@ tag_prefix = "rel-"
 addopts = "--durations=50"
 testpaths = "tests/"
 
-[tool.pylint]
-max-line-length = 88
-
-[tool.pylint.messages_control]
-disable = ["C0330", "C0326"]
-
-
 [tool.ruff]
+line-length = 88
+exclude = ["doc/", "pytensor/_version.py", "bin/pytensor_cache.py"]
+
+[tool.ruff.lint]
 select = ["C", "E", "F", "I", "UP", "W", "RUF"]
 ignore = ["C408", "C901", "E501", "E741", "RUF012"]
-exclude = ["doc/", "pytensor/_version.py", "bin/pytensor_cache.py"]
 
 
-[tool.ruff.isort]
+[tool.ruff.lint.isort]
 lines-after-imports = 2
 
 [tool.ruff.per-file-ignores]

pytensor/graph/destroyhandler.py

@@ -234,9 +234,11 @@ def fast_inplace_check(fgraph, inputs):
     """
     Supervisor = pytensor.compile.function.types.Supervisor
     protected_inputs = [
-        f.protected for f in fgraph._features if isinstance(f, Supervisor)
+        inpt
+        for f in fgraph._features
+        if isinstance(f, Supervisor)
+        for inpt in f.protected
     ]
-    protected_inputs = sum(protected_inputs, [])  # flatten the list
     protected_inputs.extend(fgraph.outputs)
 
     inputs = [

pytensor/scalar/basic.py

@@ -4096,10 +4096,11 @@ def c_code_cache_version(self):
         return tuple(rval)
 
     def c_header_dirs(self, **kwargs):
-        rval = sum(
-            (subnode.op.c_header_dirs(**kwargs) for subnode in self.fgraph.toposort()),
-            [],
-        )
+        rval = [
+            x
+            for subnode in self.fgraph.toposort()
+            for x in subnode.op.c_header_dirs(**kwargs)
+        ]
         return rval
 
     def c_support_code(self, **kwargs):

pytensor/scan/rewriting.py

@@ -1067,7 +1067,9 @@ def apply(self, fgraph):
                     if client.op.destroy_map:
                         # This flattens the content of destroy_map.values()
                         # which is a list of lists
-                        inplace_inp_indices = sum(client.op.destroy_map.values(), [])
+                        inplace_inp_indices = [
+                            x for l in client.op.destroy_map.values() for x in l
+                        ]
 
                         inplace_inps = [client.inputs[i] for i in inplace_inp_indices]
                         if original_node.inputs[inp_idx] in inplace_inps:
@@ -1860,8 +1862,8 @@ def merge(self, nodes):
         # Clone the inner graph of each node independently
         for idx, nd in enumerate(nodes):
             # concatenate all inner_ins and inner_outs of nd
-            flat_inner_ins = sum(inner_ins[idx], [])
-            flat_inner_outs = sum(inner_outs[idx], [])
+            flat_inner_ins = [x for l in inner_ins[idx] for x in l]
+            flat_inner_outs = [x for l in inner_outs[idx] for x in l]
             # clone
             flat_inner_ins, flat_inner_outs = reconstruct_graph(
                 flat_inner_ins, flat_inner_outs

pytensor/scan/utils.py

@@ -765,8 +765,8 @@ def field_names(self):
     def inner_inputs(self):
         return (
             self.inner_in_seqs
-            + sum(self.inner_in_mit_mot, [])
-            + sum(self.inner_in_mit_sot, [])
+            + [x for l in self.inner_in_mit_mot for x in l]
+            + [x for l in self.inner_in_mit_sot for x in l]
             + self.inner_in_sit_sot
             + self.inner_in_shared
             + self.inner_in_non_seqs
@@ -788,7 +788,7 @@ def outer_inputs(self):
     @property
     def inner_outputs(self):
         return (
-            sum(self.inner_out_mit_mot, [])
+            [x for l in self.inner_out_mit_mot for x in l]
             + self.inner_out_mit_sot
             + self.inner_out_sit_sot
             + self.inner_out_nit_sot

pytensor/tensor/rewriting/elemwise.py

@@ -145,11 +145,11 @@ def apply(self, fgraph):
             update_outs = []
 
         protected_inputs = [
-            f.protected
+            inpt
             for f in fgraph._features
             if isinstance(f, pytensor.compile.function.types.Supervisor)
+            for inpt in f.protected
         ]
-        protected_inputs = sum(protected_inputs, [])  # flatten the list
         protected_inputs.extend(fgraph.outputs)
         for node in list(io_toposort(fgraph.inputs, fgraph.outputs)):
             op = node.op

tests/graph/test_fg.py

@@ -574,7 +574,7 @@ def test_remove_in_and_out(self):
         assert fg.outputs == [op1_out]
         assert op3_out not in fg.clients
         assert not any(
-            op3_out.owner in clients for clients in sum(fg.clients.values(), [])
+            op3_out.owner in clients for l in fg.clients.values() for clients in l
         )
 
         # Remove an input
@@ -585,7 +585,7 @@ def test_remove_in_and_out(self):
         assert fg.inputs == [var2]
         assert fg.outputs == []
         assert not any(
-            op1_out.owner in clients for clients in sum(fg.clients.values(), [])
+            op1_out.owner in clients for l in fg.clients.values() for clients in l
         )
 
     def test_remove_duplicates(self):
@@ -622,10 +622,10 @@ def test_remove_output_empty(self):
         assert not fg.apply_nodes
         assert op1_out not in fg.clients
         assert not any(
-            op1_out.owner in clients for clients in sum(fg.clients.values(), [])
+            op1_out.owner in clients for l in fg.clients.values() for clients in l
         )
         assert not any(
-            op3_out.owner in clients for clients in sum(fg.clients.values(), [])
+            op3_out.owner in clients for l in fg.clients.values() for clients in l
         )
 
     def test_remove_node_multi_out(self):