Subquery execution under CometTakeOrderedAndProjectExec fails

[viirya]

Describe the bug

Found this bug while fixing Spark SQL test failures for #651.

The test SPARK-45584: subquery execution should not fail with ORDER BY and LIMIT in Spark SubquerySuite fails:

2024-07-31T19:48:39.7720341Z ESC[0m[ESC[0mESC[0minfoESC[0m] ESC[0mESC[0mESC[31m  org.apache.spark.SparkException: Job aborted due to stage failure: Task 0 in stage 1756.0 failed 1 times, most recent failure: Lost task 0.0 in stage 1756.0 (TID 2356) (c4b6158180c1 executor driver): org.apache.comet.CometNativeException: org.apache.comet.CometRuntimeException: Subquery 350316 not found for plan 4475.ESC[0mESC[0m

Steps to reproduce

No response

Expected behavior

No response

Additional context

No response