diagnostic flake8

gusto/diagnostics.py

@@ -1032,18 +1032,18 @@ def setup(self, domain, state_fields):
         if not hasattr(state_fields, self.field_name2):
             field1 = state_fields(self.field_name1)
             field2 = state_fields(self.field_name2, space=field1.function_space(),
-                                  pick_up=True, dump=False)         
+                                  pick_up=True, dump=False)
             # Attach state fields to self so that we can pick it up in compute
             self.state_fields = state_fields
             # The initial value for fields may not have already been set yet so we
             # postpone setting it until the compute method is called
             self.init_field_set = False
-        else: 
+        else:
             field1 = state_fields(self.field_name1)
             field2 = state_fields(self.field_name2, space=field1.function_space(),
                                   pick_up=True, dump=False)
             # By default set this new field to the current value
-            # This may be overwritten if picking up from a checkpoint 
+            # This may be overwritten if picking up from a checkpoint
             field2.assign(field1)
             self.state_fields = state_fields
             self.init_field_set = True
@@ -1101,6 +1101,7 @@ def setup(self, domain, state_fields):
 
         super().setup(domain, state_fields)
 
+
 # TODO: unify thermodynamic diagnostics
 class ThermodynamicDiagnostic(DiagnosticField):
     """Base thermodynamic diagnostic field, computing many common fields."""