Type function.copy method parameters

Author: Armavica

pytensor/compile/function/types.py

@@ -553,11 +553,11 @@ def __copy__(self):
 
     def copy(
         self,
-        share_memory=False,
-        swap=None,
-        delete_updates=False,
-        name=None,
-        profile=None,
+        share_memory: bool = False,
+        swap: dict | None = None,
+        delete_updates: bool = False,
+        name: str | None = None,
+        profile: bool | str | None = None,
     ):
         """
         Copy this function. Copied function will have separated maker and
@@ -584,7 +584,7 @@ def copy(
             If provided, will be the name of the new
             Function. Otherwise, it will be old + " copy"
 
-        profile :
+        profile : bool | str
             as pytensor.function profile parameter
 
         Returns
@@ -725,10 +725,7 @@ def checkSV(sv_ori, sv_rpl):
             profile = config.profile or config.print_global_stats
             # profile -> True or False
         if profile is True:
-            if name:
-                message = name
-            else:
-                message = str(profile.message) + " copy"
+            message = name if name else f"{profile.message} copy"
             profile = pytensor.compile.profiling.ProfileStats(message=message)
             # profile -> object
         elif isinstance(profile, str):