Fix flake8 error

3 jobs for add_density
in 2 minutes and 19 seconds, using 0 compute credits, and was queued for 1 second