def CleanupGraph(self, cleanup=False):
    """
    if cleanup, return the graph with just the cleanup transitions
    """
    if cleanup:
      graph = [(current,(action,args,result),next)
                for (current,(action,args,result),next) in self.graph 
                if self.module.cleanup in action] 
    else:
      graph = self.graph
    return graph
dataset/ETHPy150Open jon-jacky/PyModel/pymodel/FSM.py/FSM.CleanupGraph/swapped operands of `in`