Add boilerplate code

This commit is contained in:
Simon Alibert
2024-05-20 19:16:37 +02:00
parent c4da689171
commit 512df4b468
8 changed files with 240 additions and 3 deletions

17
gym_dora/example.py Normal file
View File

@@ -0,0 +1,17 @@
import gymnasium as gym
import gym_dora # noqa: F401
env = gym.make("gym_dora/DoraAloha-v0", disable_env_checker=True)
obs = env.reset()
policy = ... # make_policy
done = False
while not done:
actions = policy.select_action(obs)
observation, reward, terminated, truncated, info = env.step(actions)
done = terminated | truncated | done
env.close()