Gym environment scripts

2023-12-20 20:43:52 -05:00
parent 3b072fc196
commit b95f5e96ec
3 changed files with 459 additions and 0 deletions
--- a/rotorpy/learning/init.py
+++ b/rotorpy/learning/init.py
@@ -0,0 +1,6 @@
+from gymnasium.envs.registration import register
+
+register(
+     id="Quadrotor-v0",
+     entry_point="rotorpy.learning.quadrotor_environments:QuadrotorEnv",
+)