Merge branch 'main' into uv-installable

delvtech · Mar 5, 2024 · 661698c · 661698c
2 parents fac56f3 + 719ea8d
commit 661698c
Show file tree

Hide file tree

Showing 7 changed files with 606 additions and 12 deletions.
diff --git a/.gitignore b/.gitignore
@@ -35,6 +35,7 @@ __pypackages__/
 
 # Default traiderdaive model output
 .traider_models
+.saved_models
 
 # build directories
 .build

diff --git a/lib/traiderdaive/pyproject.toml b/lib/traiderdaive/pyproject.toml
@@ -23,7 +23,8 @@ with-dependencies = [
 ]
 base = [
     "gymnasium",
-    "stable-baselines3"
+    "stable-baselines3",
+    "scipy",
 ]
 lateral = [
     # Lateral dependencies across subpackages

diff --git a/lib/traiderdaive/traiderdaive/__init__.py b/lib/traiderdaive/traiderdaive/__init__.py
@@ -2,11 +2,18 @@
 
 from gymnasium.envs.registration import register
 
+from .gym_environments.full_hyperdrive_env import FullHyperdriveEnv
 from .gym_environments.simple_hyperdrive_env import SimpleHyperdriveEnv
 
-# Register simple hyperdrive env to gym
+# Register hyperdrive envs to gym
 register(
     id="traiderdaive/simple_hyperdrive_env",
     entry_point="traiderdaive.gym_environments.simple_hyperdrive_env:SimpleHyperdriveEnv",
-    max_episode_steps=300,
+    max_episode_steps=1000,
+)
+
+register(
+    id="traiderdaive/full_hyperdrive_env",
+    entry_point="traiderdaive.gym_environments.full_hyperdrive_env:FullHyperdriveEnv",
+    max_episode_steps=1000,
 )