Not urgent but would appreciate a flyte review on this as well: https://github.com/flyteorg/flyte/pull/6485