fix: move check_inputs if available during to_torchscript.

Author: GdoongMathew

src/lightning/pytorch/core/module.py

@@ -1473,8 +1473,8 @@ def forward(self, x):
                 example_inputs = self.example_input_array
 
             if kwargs.get("check_inputs") is not None:
-                check_inputs = self._on_before_batch_transfer(kwargs["check_inputs"])
-                kwargs["check_inputs"] = self._apply_batch_transfer_handler(check_inputs)
+                kwargs["check_inputs"] = self._on_before_batch_transfer(kwargs["check_inputs"])
+                kwargs["check_inputs"] = self._apply_batch_transfer_handler(kwargs["check_inputs"])
 
             # automatically send example inputs to the right device and use trace
             example_inputs = self._on_before_batch_transfer(example_inputs)