Hi @Kevin Su , I left a comment here https://github.com/flyteorg/flyteplugins/pull/398#issuecomment-1742682067 regarding this change. Could you please let me know what you think? Thank you.

cc @Yee / @Haytham Abuelfutuh can you folks take a look too