To generate parquet tables from a Grid2op environment:
frompathlibimportPathfromlightsim2gridimportLightSimBackendimportgrid2opfromgrid2op.AgentimportRandomAgentfromgrid2op.Environment.EnvRecorderimportEnvRecorderif__name__=="__main__":
env=grid2op.make(
"<PATH TO ENV>",
test=True,
backend=LightSimBackend(),
_add_to_name="test"
)
withEnvRecorder(env, Path('<PATH TO RECORDED DATA>')) asenv_rec:
obs=env_rec.reset()
do_nothing=env.action_space()
reward=0done=Falsetotal_reward=0agent=RandomAgent(env.action_space)
whilenotdone:
action=agent.act(obs, reward, done)
obs, reward, done, info=env_rec.step(action)