Apply rule UP031

Author: Armavica

pytensor/compile/profiling.py

@@ -514,14 +514,12 @@ def summary_class(self, file=sys.stderr, N=None):
             # While this carries over less information, it is arranged such
             # that it is way more readable than the previous output of the
             # profiler
+        nb_classes = max(0, len(otimes) - N)
+        percent_total = sum(f for f, t, a, ci, nb_call, nb_op in otimes[N:])
+        time_total = sum(t for f, t, a, ci, nb_call, nb_op in otimes[N:])
         print(
-            "   ... (remaining %i Classes account for %6.2f%%(%.2fs) of "
-            "the runtime)"
-            % (
-                max(0, len(otimes) - N),
-                sum(f for f, t, a, ci, nb_call, nb_op in otimes[N:]),
-                sum(t for f, t, a, ci, nb_call, nb_op in otimes[N:]),
-            ),
+            f"   ... (remaining {nb_classes} Classes account for "
+            f"{percent_total:6.2f}%%({time_total:.2f}s) of the runtime)",
             file=file,
         )
         print("", file=file)
@@ -607,14 +605,12 @@ def summary_ops(self, file=sys.stderr, N=None):
             # While this carries over less information, it is arranged such
             # that it is way more readable than the previous output of the
             # profiler
+        nb_ops = max(0, len(otimes) - N)
+        percent_total = sum(f for f, t, a, ci, nb_call, nb_op in otimes[N:])
+        time_total = sum(t for f, t, a, ci, nb_call, nb_op in otimes[N:])
         print(
-            "   ... (remaining %i Ops account for %6.2f%%(%.2fs) of "
-            "the runtime)"
-            % (
-                max(0, len(otimes) - N),
-                sum(f for f, t, a, ci, nb_call, nb_op in otimes[N:]),
-                sum(t for f, t, a, ci, nb_call, nb_op in otimes[N:]),
-            ),
+            f"   ... (remaining {nb_ops} Ops account for "
+            f"{percent_total:6.2f}%%({time_total:.2f}s) of the runtime)",
             file=file,
         )
         print("", file=file)

pytensor/link/c/basic.py

@@ -108,12 +108,12 @@ def failure_code(sub, use_goto=True):
         be careful to avoid executing incorrect code.
 
     """
+    id = sub["id"]
+    failure_var = sub["failure_var"]
     if use_goto:
-        goto_statement = "goto __label_%(id)i;" % sub
+        goto_statement = f"goto __label_{id};"
     else:
         goto_statement = ""
-    id = sub["id"]
-    failure_var = sub["failure_var"]
     return f"""{{
         {failure_var} = {id};
         if (!PyErr_Occurred()) {{

tests/tensor/rewriting/test_elemwise.py

@@ -1076,7 +1076,7 @@ def test_elemwise_fusion(self, case, nb_repeat=1, assert_len_topo=True):
 
     def test_fusion_35_inputs(self):
         r"""Make sure we don't fuse too many `Op`\s and go past the 31 function arguments limit."""
-        inpts = vectors(["i%i" % i for i in range(35)])
+        inpts = vectors([f"i{i}" for i in range(35)])
 
         # Make an elemwise graph looking like:
         # sin(i34 + sin(i33 + sin(... i1 + sin(i0) ...)))