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