swangnice/rotor_py_control
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Made quadrotor gymnasium env compatible with Gymnasium 1.0.0 (Issue #11)

Browse Source
This commit is contained in:
spencerfolk
2025-01-20 12:32:43 -05:00
parent 8500cad9fa
commit 68ad764727
4 changed files with 35 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/ppo_hover_train.py
View File

@@ -59,7 +59,7 @@ env = gym.make("Quadrotor-v0",
# check_env(env, warn=True) # you can check the environment using built-in tools
# Reset the environment
observation, info = env.reset(initial_state='random', options={'pos_bound': 2, 'vel_bound': 0})
observation, info = env.reset(options={'initial_state': 'random', 'pos_bound': 2, 'vel_bound': 0})
# Create a new model
model = PPO(MlpPolicy, env, verbose=1, ent_coef=0.01, tensorboard_log=log_dir)