Add support for python 3.11.

google-research · Jun 3, 2023 · 93cb7d7 · 93cb7d7
1 parent 2d67aaa
commit 93cb7d7
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -40,9 +40,7 @@ We've created an introductory [Colab](https://colab.research.google.com/github/g
 
 ## Installation
 
-RoboPianist is supported on both Linux and macOS and can be installed with Python 3.8 up to 3.10. We recommend using [Miniconda](https://docs.conda.io/en/latest/miniconda.html) to manage your Python environment.
-
-**3.11 will be supported once the numba team resolves [#8304](https://github.com/numba/numba/issues/8304).**
+RoboPianist is supported on both Linux and macOS and can be installed with Python >= 3.8. We recommend using [Miniconda](https://docs.conda.io/en/latest/miniconda.html) to manage your Python environment.
 
 ### Install from source
 

diff --git a/robopianist/__init__.py b/robopianist/__init__.py
@@ -14,7 +14,7 @@
 
 from pathlib import Path
 
-__version__ = "1.0.8"
+__version__ = "1.0.9"
 
 # Path to the root of the project.
 _PROJECT_ROOT = Path(__file__).parent.parent

diff --git a/robopianist/suite/tasks/piano_with_one_shadow_hand.py b/robopianist/suite/tasks/piano_with_one_shadow_hand.py
@@ -238,7 +238,7 @@ def _compute_key_press_reward(self, physics) -> float:
             rew += 0.5 * rews.mean()
         # If there's any false positive, the remaining 0.5 reward is lost.
         off = np.flatnonzero(1 - self._goal_current[:-1])
-        rew += 0.5 * (1 - self.piano.activation[off].any())
+        rew += 0.5 * (1 - float(self.piano.activation[off].any()))
         return rew
 
     def _compute_fingering_reward(self, physics) -> float:

diff --git a/robopianist/suite/tasks/piano_with_shadow_hands.py b/robopianist/suite/tasks/piano_with_shadow_hands.py
@@ -288,7 +288,7 @@ def _compute_key_press_reward(self, physics: mjcf.Physics) -> float:
             rew += 0.5 * rews.mean()
         # If there are any false positives, the remaining 0.5 reward is lost.
         off = np.flatnonzero(1 - self._goal_current[:-1])
-        rew += 0.5 * (1 - self.piano.activation[off].any())
+        rew += 0.5 * (1 - float(self.piano.activation[off].any()))
         return rew
 
     def _compute_fingering_reward(self, physics: mjcf.Physics) -> float:

diff --git a/robopianist/viewer/views.py b/robopianist/viewer/views.py
@@ -234,7 +234,9 @@ class TimeSeries:
             sequences of floats. Will be used to annotate the legend.
     """
 
-    data: Deque[Union[float, Sequence[float]]] = deque(maxlen=mujoco.mjMAXLINE)
+    data: Deque[Union[float, Sequence[float]]] = field(
+        default_factory=lambda: deque(maxlen=mujoco.mjMAXLINE)
+    )
     linename: List[str] = field(default_factory=list)
 
     def add(self, value: float) -> None:

diff --git a/setup.py b/setup.py
@@ -68,6 +68,7 @@
     "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
 ]
 
@@ -94,7 +95,7 @@
     license="Apache License 2.0",
     license_files=("LICENSE",),
     packages=find_packages(exclude=["examples"]),
-    python_requires=">=3.8, <3.11",
+    python_requires=">=3.8",
     install_requires=core_requirements,
     include_package_data=True,
     classifiers=classifiers,