Pull constant values out of loop and check earlier

[elharo]

Description

Pull constant values out of loop and check for $row_id earlier

Motivation and Context

Debugging weird failures in prod. Code works in some clusters and fails in others.

Impact

none

Test Plan

CI

Contributor checklist

• Please make sure your submission complies with our development, formatting, commit message, and attribution guidelines.
• PR description addresses the issue accurately and concisely. If the change is non-trivial, a GitHub Issue is referenced.
• Documented new properties (with its default value), SQL syntax, functions, or other functionality.
• If release notes are required, they follow the release notes guidelines.
• Adequate tests were added if applicable.
• CI passed.

Release Notes

== NO RELEASE NOTE ==

[sdruzkin]

LGTM