Fix unpacking

Co-authored-by: Ricardo Vieira <[email protected]>

Author: 2 people

pytensor/link/pytorch/dispatch/scalar.py

@@ -80,10 +80,7 @@ def scalar_loop(steps, *start_and_constants):
                 *carry, done = update(*carry, *constants)
                 if torch.any(done):
                     break
-            if len(node.outputs) == 2:
-                return carry[0], done
-            else:
-                return carry, done
+           return *carry, done
     else:
 
         def scalar_loop(steps, *start_and_constants):