Question How to align texture to a Flowmap?

Hello! I've been playing with Flowmaps and want to align the texture in the direction of the Flowmaster.

Catlikecoding does this with script in his blog:

https://catlikecoding.com/unity/tutorials/flow/directional-flow/

I'm trying to recreate just this aspect of a flow shader with shader graph... after several days of beating my head against the keyboard I'm no further ahead.

Anyone have a solution to align uv's in the direction of the Flowmap?

Thanks!!