Merge remote-tracking branch 'origin/dev' into feature/2628-update-benchmarking-script-branched

2024-06-01 00:20:21 +01:00
parent 5f02846879 c705a15d00
commit 472040aa70
35 changed files with 427 additions and 348 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,11 +36,8 @@ dependencies = [
    "polars==0.20.30",
    "prettytable==3.8.0",
    "PyYAML==6.0",
-    "stable-baselines3[extra]==2.1.0",
-    "tensorflow==2.12.0",
    "typer[all]==0.9.0",
    "pydantic==2.7.0",
-    "ray[rllib] >= 2.9, < 3",
    "ipywidgets",
    "deepdiff"
 ]
@@ -56,6 +53,11 @@ license-files = ["LICENSE"]


 [project.optional-dependencies]
+rl = [
+    "ray[rllib] >= 2.20.0, < 3",
+    "tensorflow==2.12.0",
+    "stable-baselines3[extra]==2.1.0",
+]
 dev = [
    "build==0.10.0",
    "flake8==6.0.0",