- Modify sncps/models/cnp.py to accept keras model as arg for CNP - See [TF example for GAN](https://www.tensorflow.org/guide/keras/customizing_what_happens_in_fit#wrapping_up_an_end-to-end_gan_example), they do something similar.