diff --git a/grits/coarsegrain.py b/grits/coarsegrain.py index 7c249e3..bcd691b 100644 --- a/grits/coarsegrain.py +++ b/grits/coarsegrain.py @@ -727,7 +727,7 @@ def save(self, cg_gsdfile, start=0, stop=None, stride=1): new_snap = gsd.hoomd.Frame() position = [] mass = [] - #make an empty list for forces here + # make an empty list for forces here orientation = [] if self.aniso_beads else None f_box = freud.Box.from_box(s.configuration.box) @@ -741,8 +741,8 @@ def save(self, cg_gsdfile, start=0, stop=None, stride=1): sum(s.particles.mass[x]) * self.mass_scale for x in inds ] - #do the force calculation here - + # do the force calculation here + if self.aniso_beads: for x in inds: masses = s.particles.mass[x] * self.mass_scale