Cartpole_lstm.py example fails

Hello, I have 2 docker containers, one with ray 1.7.0 installed and one with ray 2.0.0.dev0 nightly build.

1.7.0 :

2.0.0.dev0 :

they both train but I have two different error at the end.