Fix: examples

Author: TheEimer

arlbench/__init__.py

@@ -7,7 +7,7 @@
 
 __author__ = """AutoML Hannover"""
 __email__ = """automl@ai.uni-hannover.de"""
-__version__ = """0.1.0"""
+__version__ = """0.1.1"""
 __copyright__ = f"Copyright {datetime.date.today().strftime('%Y')}, AutoML Hannover"
 
 __all__ = ["AutoRLEnv", "run_arlbench"]

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

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

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

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"]