Merge pull request #19 from LegionAtol/ideas

@LegionAtols GSOC RL algorithm
qutip · Oct 4, 2024 · d3ce1de · d3ce1de
2 parents 9d6ef32 + 604b12e
commit d3ce1de
Show file tree

Hide file tree

Showing 6 changed files with 462 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -92,3 +92,5 @@ repos:
           - jaxlib
           - diffrax
           - pytest
+          - gymnasium
+          - stable-baselines3
diff --git a/requirements.txt b/requirements.txt
@@ -7,3 +7,5 @@ qutip>=5.0.1
 qutip-qtrl
 qutip-jax
 pre-commit
+gymnasium>=0.29.1
+stable-baselines3>=2.3.2
diff --git a/setup.cfg b/setup.cfg
@@ -35,6 +35,8 @@ install_requires =
     qutip-qtrl
     qutip-jax
     numpy>=1.16.6,<2.0
+    gymnasium>=0.29.1
+    stable-baselines3>=2.3.2
 setup_requires =
     cython>=1.0
     packaging
-Original file line number
+Diff line change
@@ Expand Up / @@ -92,3 +92,5 @@ repos: @@
               - jaxlib
               - diffrax
               - pytest
+              - gymnasium
+              - stable-baselines3