fix flake8

theano/gof/fg.py

@@ -534,6 +534,7 @@ class FunctionGraph(utils.object2):
         # transaction will be reverted later.
         self.execute_callbacks('on_change_input', node, i,
                                r, new_r, reason=reason)
+        if prune:
             self.__remove_clients__(r, [], True)
 
     # replace #