@David Espejo (he/him) / @jeev can either of you take a look at this PR https://github.com/flyteorg/flyte/pull/4337/files