Fix requirements

RobertTLange · Apr 21, 2024 · 88fef69 · 88fef69
1 parent 221c3cc
commit 88fef69
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+### [v0.0.8] - 21/04/2024
+
+Fix import errors for `matplotlib` and `seaborn` visualization.
+
 ### [v0.0.7] - 04/04/2024
 
 1. Fixed most of the typing issues under pylint --strict.

diff --git a/gymnax/_version.py b/gymnax/_version.py
@@ -1,3 +1,3 @@
 """Version file."""
 
-__version__ = "0.0.7"
+__version__ = "0.0.8"
diff --git a/setup.py b/setup.py
@@ -31,7 +31,17 @@ def parse_requirements(path: str) -> List[str]:
     raise RuntimeError("Unable to find version string in %s." % (VERSIONFILE,))
 git_tar = f"https://github.com/RobertTLange/gymnax/archive/v{verstr}.tar.gz"
 
-requires = ["jax", "jaxlib", "chex", "flax", "pyyaml", "gym>=0.26", "gymnasium"]
+requires = [
+    "jax",
+    "jaxlib",
+    "chex",
+    "flax",
+    "pyyaml",
+    "gym>=0.26",
+    "gymnasium",
+    "matplotlib",
+    "seaborn",
+]
 test_requires = ["bsuite", "minatar", "brax"]
 
 setup(