ray v1.0.0, python 3.7, running ray locally for the below experiment.
Following the documentation at:
https://docs.ray.io/en/latest/tune/api_docs/trainable.htm