Fix mypy signature

Author: Ian Schweer

pytensor/link/pytorch/dispatch/scalar.py

@@ -71,8 +71,7 @@ def scalar_loop(steps, *start_and_constants):
             return torch.stack(carry), torch.tensor([done])
     else:
 
-        def scalar_loop(*args):
-            steps, *start_and_constants = args
+        def scalar_loop(steps, *start_and_constants):
             carry, constants = (
                 start_and_constants[:state_length],
                 start_and_constants[state_length:],