Merge branch 'instadeepai:main' into boxoban

instadeepai · Jan 2, 2024 · 4ee995f · 4ee995f
2 parents 2913197 + e8d51e0
commit 4ee995f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,9 @@
 <p align="center">
-    <a href="docs/img/jumanji_logo.png">
-        <img src="docs/img/jumanji_logo.png" alt="Jumanji logo" width="50%"/>
-    </a>
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="docs/img/jumanji_logo_dm.png">
+  <source media="(prefers-color-scheme: light)" srcset="docs/img/jumanji_logo.png">
+  <img alt="Jumanji Logo" src="docs/img/jumanji_logo.png", width="50%">
+</picture>
 </p>
 
 [![Python Versions](https://img.shields.io/pypi/pyversions/jumanji.svg?style=flat-square)](https://www.python.org/doc/versions/)

diff --git a/docs/img/jumanji_logo_dm.png b/docs/img/jumanji_logo_dm.png
diff --git a/jumanji/wrappers.py b/jumanji/wrappers.py
@@ -96,6 +96,14 @@ def action_spec(self) -> specs.Spec:
         """Returns the action spec."""
         return self._env.action_spec()
 
+    def reward_spec(self) -> specs.Array:
+        """Returns the reward spec."""
+        return self._env.reward_spec()
+
+    def discount_spec(self) -> specs.BoundedArray:
+        """Returns the discount spec."""
+        return self._env.discount_spec()
+
     def render(self, state: State) -> Any:
         """Compute render frames during initialisation of the environment.