python code changes per cellrank v2

[josef0731]

As per #11. Revise python code to make sure syntax works. In the mean time, need to specify in conda environment specification to not install cellrank v2.