Apply rule RUF059

Author: Armavica

pytensor/compile/builders.py

@@ -122,7 +122,7 @@ def construct_nominal_fgraph(
     (
         local_inputs,
         local_outputs,
-        (clone_d, update_d, update_expr, new_shared_inputs),
+        (_clone_d, update_d, update_expr, new_shared_inputs),
     ) = new
 
     assert len(local_inputs) == len(inputs) + len(implicit_shared_inputs)

pytensor/compile/function/pfunc.py

@@ -591,7 +591,7 @@ def construct_pfunc_ins_and_outs(
             clone_inner_graphs=True,
         )
         input_variables, cloned_extended_outputs, other_stuff = output_vars
-        clone_d, update_d, update_expr, shared_inputs = other_stuff
+        clone_d, update_d, _update_expr, shared_inputs = other_stuff
 
         # Recover only the clones of the original outputs
         if outputs is None:

pytensor/compile/mode.py

@@ -407,7 +407,7 @@ def register(self, *optimizations):
             optimizations.
         """
 
-        link, opt = self.get_linker_optimizer(
+        _link, opt = self.get_linker_optimizer(
             self.provided_linker, self.provided_optimizer
         )
         return self.clone(optimizer=opt.register(*optimizations))

pytensor/compile/profiling.py

@@ -1323,7 +1323,7 @@ def print_stats(stats1, stats2):
         )
         print_stats(stats[1], stats[3])
 
-        (max_node_memory_size, _, _, _) = stats[0]
+        (_max_node_memory_size, _, _, _) = stats[0]
         (new_max_node_memory_size, _, _, _) = stats[2]
         print(
             "    Max peak memory if allow_gc=False (linker don't make a difference)",

pytensor/graph/fg.py

@@ -559,7 +559,7 @@ def remove_node(self, node: Apply, reason: str | None = None):
 
             out_clients = clients.get(out, ())
             while out_clients:
-                out_client, out_idx = out_clients.pop()
+                out_client, _out_idx = out_clients.pop()
 
                 if isinstance(out_client.op, Output):
                     self.remove_output(out_client.op.idx, remove_client=False)

pytensor/graph/rewriting/basic.py

@@ -2827,7 +2827,7 @@ def local_recursive_function(
     else:
         out_index = 0
 
-    final_outs, rewritten_nodes = local_recursive_function(rewrites, out, {}, 0)
+    final_outs, _rewritten_nodes = local_recursive_function(rewrites, out, {}, 0)
     return final_outs[out_index]
 
 

pytensor/link/basic.py

@@ -509,7 +509,7 @@ def make_thunk(self, **kwargs):
         kwargs.pop("input_storage", None)
         make_all += [x.make_all(**kwargs) for x in self.linkers[1:]]
 
-        fns, input_lists, output_lists, thunk_lists, order_lists = zip(
+        _fns, input_lists, output_lists, thunk_lists, order_lists = zip(
             *make_all, strict=True
         )
 

pytensor/link/c/basic.py

@@ -1738,7 +1738,7 @@ def find_task(self, failure_code):
     def __call__(self):
         failure = self.run_cthunk(self.cthunk)
         if failure:
-            task, taskname, id = self.find_task(failure)
+            task, _taskname, _id = self.find_task(failure)
             try:
                 trace = task.trace
             except AttributeError:
@@ -1980,7 +1980,7 @@ def make_thunk(self, **kwargs):
             .make_all(**kwargs)
         )
         kwargs.pop("input_storage", None)
-        _f, i2, o2, thunks2, order2 = (
+        _f, i2, _o2, thunks2, order2 = (
             OpWiseCLinker(schedule=self.schedule)
             .accept(fgraph, no_recycling=no_recycling)
             .make_all(**kwargs)

pytensor/link/c/cmodule.py

@@ -1496,7 +1496,7 @@ def clear_unversioned(self, min_age=None):
                 # May happen for broken versioned keys.
                 continue
             for key_idx, key in enumerate(key_data.keys):
-                version, rest = key
+                version, _rest = key
                 if version:
                     # Since the version is included in the module hash,
                     # it should not be possible to mix versioned and
@@ -2243,7 +2243,7 @@ def join_options(init_part):
                                 if "target-cpu" in p:
                                     opt = p.split()
                                     if len(opt) == 2:
-                                        opt_name, opt_val = opt
+                                        _opt_name, opt_val = opt
                                         new_flags[i] = f"-march={opt_val}"
 
                             # Some versions of GCC report the native arch
@@ -2474,7 +2474,7 @@ def linking_patch(lib_dirs: list[str], libs: list[str]) -> list[str]:
         else:
             # In explicit else because of https://github.com/python/mypy/issues/10773
             def sort_key(lib):
-                name, *numbers, extension = lib.split(".")
+                _name, *numbers, extension = lib.split(".")
                 return (extension == "dll", tuple(map(int, numbers)))
 
             patched_lib_ldflags = []

pytensor/link/c/op.py

@@ -84,7 +84,7 @@ def is_f16(t):
         outputs = cl.make_thunk(
             input_storage=node_input_storage, output_storage=node_output_storage
         )
-        thunk, node_input_filters, node_output_filters = outputs
+        thunk, _node_input_filters, _node_output_filters = outputs
 
         if compute_map is None:
             rval = is_cthunk_wrapper_type(thunk)