Finish primary refactor. Add tests.

utiasDSL · Jun 23, 2024 · b001302 · b001302
1 parent 0a9f16d
commit b001302
Show file tree

Hide file tree

Showing 36 changed files with 1,529 additions and 51,171 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,10 +6,11 @@ authors = ["Your Name <[email protected]>"]
 license = "MIT"
 
 [tool.poetry.dependencies]
+# TODO: Clean up dependencies
 python = "^3.8.10"
 matplotlib = "^3.5.1"
 Pillow = "^9.0.0"
-munch = "^2.5.0"
+munch = "^4.0.0"
 PyYAML = "^6.0"
 imageio = "^2.14.1"
 dict-deep = "^4.1.2"

diff --git a/safe_control_gym/__init__.py b/safe_control_gym/__init__.py
@@ -1 +1 @@
-from safe_control_gym.envs import *
+import safe_control_gym.envs  # noqa: F401, load envs into gymnasium registry
diff --git a/safe_control_gym/envs/__init__.py b/safe_control_gym/envs/__init__.py
@@ -3,12 +3,12 @@
 from gymnasium import register
 
 register(
-    id="quadrotor",
-    entry_point="safe_control_gym.envs.quadrotor:Quadrotor",
+    id="drone_sim",
+    entry_point="safe_control_gym.envs.drone_sim:DroneSim",
 )
 
 register(
-    id="firmware",
-    entry_point="safe_control_gym.envs.firmware_wrapper:FirmwareWrapper",
+    id="drone_racing-v0",
+    entry_point="safe_control_gym.envs.drone_racing_env:DroneRacingEnv",
     max_episode_steps=900,  # 30 seconds * 30 Hz
 )