Replace static joins with f-strings

Author: Armavica

pytensor/link/c/op.py

@@ -451,7 +451,7 @@ def c_init_code_apply(self, node, name):
             code = self.code_sections["init_code_apply"]
 
             define_macros, undef_macros = self.get_c_macros(node, name)
-            return "\n".join(["", define_macros, code, undef_macros])
+            return f"\n{define_macros}\n{code}\n{undef_macros}"
         else:
             return super().c_init_code_apply(node, name)
 
@@ -460,7 +460,7 @@ def c_support_code_apply(self, node, name):
             code = self.code_sections["support_code_apply"]
 
             define_macros, undef_macros = self.get_c_macros(node, name)
-            return "\n".join(["", define_macros, code, undef_macros])
+            return f"\n{define_macros}\n{code}\n{undef_macros}"
         else:
             return super().c_support_code_apply(node, name)
 
@@ -469,7 +469,7 @@ def c_support_code_struct(self, node, name):
             code = self.code_sections["support_code_struct"]
 
             define_macros, undef_macros = self.get_c_macros(node, name)
-            return "\n".join(["", define_macros, code, undef_macros])
+            return f"\n{define_macros}\n{code}\n{undef_macros}"
         else:
             return super().c_support_code_struct(node, name)
 
@@ -478,7 +478,7 @@ def c_cleanup_code_struct(self, node, name):
             code = self.code_sections["cleanup_code_struct"]
 
             define_macros, undef_macros = self.get_c_macros(node, name)
-            return "\n".join(["", define_macros, code, undef_macros])
+            return f"\n{define_macros}\n{code}\n{undef_macros}"
         else:
             return super().c_cleanup_code_struct(node, name)
 
@@ -561,9 +561,7 @@ def c_init_code_struct(self, node, name, sub):
             def_macros, undef_macros = self.get_c_macros(node, name)
             def_sub, undef_sub = get_sub_macros(sub)
 
-            return "\n".join(
-                ["", def_macros, def_sub, op_code, undef_sub, undef_macros]
-            )
+            return f"\n{def_macros}\n{def_sub}\n{op_code}\n{undef_sub}\n{undef_macros}"
         else:
             return super().c_init_code_struct(node, name, sub)
 
@@ -597,16 +595,9 @@ def c_code(self, node, name, inp, out, sub):
                 def_sub, undef_sub = get_sub_macros(sub)
                 def_io, undef_io = get_io_macros(inp, out)
 
-                return "\n".join(
-                    [
-                        def_macros,
-                        def_sub,
-                        def_io,
-                        op_code,
-                        undef_io,
-                        undef_sub,
-                        undef_macros,
-                    ]
+                return (
+                    f"{def_macros}\n{def_sub}\n{def_io}\n{op_code}"
+                    f"\n{undef_io}\n{undef_sub}\n{undef_macros}"
                 )
             else:
                 raise NotImplementedError()
@@ -620,16 +611,9 @@ def c_code_cleanup(self, node, name, inputs, outputs, sub):
             def_sub, undef_sub = get_sub_macros(sub)
             def_io, undef_io = get_io_macros(inputs, outputs)
 
-            return "\n".join(
-                [
-                    def_macros,
-                    def_sub,
-                    def_io,
-                    op_code,
-                    undef_io,
-                    undef_sub,
-                    undef_macros,
-                ]
+            return (
+                f"{def_macros}\n{def_sub}\n{def_io}\n{op_code}"
+                f"\n{undef_io}\n{undef_sub}\n{undef_macros}"
             )
         else:
             return super().c_code_cleanup(node, name, inputs, outputs, sub)

pytensor/tensor/elemwise_cgen.py

@@ -440,9 +440,7 @@ def get_loop_strides(loop_order, i):
         }} // end loop {int(i)}
         """
 
-    return "\n".join(
-        ["{", order_loops, declare_totals, declare_strides, declare_iter, loop, "}\n"]
-    )
+    return f"{{\n{order_loops}\n{declare_totals}\n{declare_strides}\n{declare_iter}\n{loop}\n}}\n"
 
 
 # print make_declare(((0, 1, 2, 3), ('x', 1, 0, 3), ('x', 'x', 'x', 0)),