Fix: examples

automl · Jun 5, 2024 · 29ba718 · 29ba718
1 parent 11be1f6
commit 29ba718
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 3 deletions.
diff --git a/arlbench/__init__.py b/arlbench/__init__.py
@@ -7,7 +7,7 @@
 
 __author__ = """AutoML Hannover"""
 __email__ = """[email protected]"""
-__version__ = """0.1.0"""
+__version__ = """0.1.1"""
 __copyright__ = f"Copyright {datetime.date.today().strftime('%Y')}, AutoML Hannover"
 
 __all__ = ["AutoRLEnv", "run_arlbench"]
diff --git a/examples/configs/algorithm/dqn.yaml b/examples/configs/algorithm/dqn.yaml
@@ -9,6 +9,7 @@ hp_config:
   buffer_size: 1000000
   initial_epsilon: 1.0
   target_epsilon: 0.05
+  exploration_fraction: 0.1
   gamma: 0.99
   gradient_steps: 1
   learning_rate: 0.0003

diff --git a/examples/configs/epsilon_heuristic.yaml b/examples/configs/epsilon_heuristic.yaml
@@ -45,6 +45,7 @@ hp_config:
   buffer_size: 1000000
   initial_epsilon: 0.9
   target_epsilon: 0.9
+  exploration_fraction: 0.1
   gamma: 0.99
   gradient_steps: 1
   learning_rate: 0.0003

diff --git a/examples/configs/gradient_lr.yaml b/examples/configs/gradient_lr.yaml
@@ -45,6 +45,7 @@ hp_config:
   buffer_size: 1000000
   initial_epsilon: 0.8
   target_epsilon: 0.8
+  exploration_fraction: 0.1
   gamma: 0.99
   gradient_steps: 1
   learning_rate: 0.0003

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "ARLBench"
-version = "0.1.0"
+version = "0.1.3"
 dependencies = [
   "typing_extensions", # Better typing
   "tomli==2.0.1", 
@@ -19,6 +19,7 @@ dependencies = [
   "brax==0.10.3",
   "pytest==7.4.4",
   "coverage==7.4.4",
+  "chex==0.1.86",
   "ruff",
   "hydra-core",
   "hydra-submitit-launcher",
@@ -54,7 +55,7 @@ tooling = ["commitizen", "pre-commit", "ruff"]
 test = ["pytest", "pytest-coverage", "pytest-cases", "ARLBench[examples]"]
 examples = ["hypersweeper"]
 doc = [
-  "automl_sphinx_theme",
+  "automl_sphinx_theme", "gymnasium==0.29.1"
 ]
 envpool = ["envpool==0.8.4"]