Update tests: change lambda to partial.

chromatix-team / chromatix

Differentiable wave optics using JAX! Documentation can be found at https://chromatix.readthedocs.io

MIT License

77 stars 13 forks source link

Update tests: change lambda to partial. #82

Closed GJBoth closed 1 year ago

GJBoth commented 1 year ago

I was planning on changing the pupils to get rid of the ugly lambda field: square_pupil(field, diameter) stuff in some of the sources but realized just changing them to a partial is much nicer. So this PR literally only does that.

diptodip commented 1 year ago

Nice :+1: