how to code ray with python