Hi @MichaelAzmy , thanks for reporting the issue. If you use the fix on master which removes the as_pydict() conversion, does this unblock your use case? It looks like we made this change as a part of this PR to enable Ray Data strict mode by default for Ray 2.5.