def write_policy_model_file(epoch): with gfile.GFile( ppo.get_policy_model_file_from_epoch(output_dir, epoch), 'w') as f: f.write('some data')
def test_get_policy_model_file_from_epoch(self): self.assertEqual('/tmp/model-000000.pkl', ppo.get_policy_model_file_from_epoch('/tmp', 0)) self.assertEqual('/tmp/model-123456.pkl', ppo.get_policy_model_file_from_epoch('/tmp', 123456))
def write_policy_model_file(epoch): fname = ppo.get_policy_model_file_from_epoch(output_dir, epoch) with gfile.GFile(fname, 'w') as f: f.write('some data') return fname