# Trace the folder

class TraceCollector():
    def __init__(self, steps: int = None):
        self.steps = [] # list of steps

    def logs(self, state, action, output):
        return self.steps.append({"state": state,
                                  "action": action,
                                  "output": output})