Update assimilation.py documentation.