Method load_data

ppo/ppo_datahelper.py:130–137

(self, file_path: str)

        self.batch_size = opt.rollout_batch_size # batch size for sampling from env
    
    def load_data(self, file_path: str):
        with open(file_path, 'r') as f:
            data: List[List[str]] = json.load(f)
            
        output: List[List[str]] = [sample for sample in data if all(sample)]
        del data

        return output
    
    def format(self, sample: List[str]) -> Dict[str, Any]:
        context = sample