ppo code running error

Hello, can you take a look at the following error? Thanks.

run code experiment.py

https://github.com/labmlai/annotated_deep_learning_paper_implementations/blob/a0679ecd90b41b8e012995a6bdf095edae590b17/labml_nn/rl/game.py#L130

cv2.error: OpenCV(4.9.0) :-1: error: (-5:Bad argument) in function 'cvtColor' Overload resolution failed:

• src is not a numerical tuple
• Expected Ptrcv::UMat for argument 'src'