File size: 336 Bytes
03a907a
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
# 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})