Hi @carlorop ,
@Lars_Simon_Zehnder has written a few lines on this before.
I am not aware of any “1 liner” solution for this in RLlib right right now.
Nevertheless, it is possible. Do you need help with coding it? This is a usecase that I am interested in and we can work on it together if you like. 