behavioural_cloning.py (1 line): - line 96: # TODO need to clean up this hidden state after worker is done with the work item.