chore: upgrade to python3.10

EdanToledo · Feb 17, 2024 · eceeab8 · eceeab8
1 parent 385cc72
commit eceeab8
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 9 deletions.
diff --git a/.github/workflows/test_linters.yaml b/.github/workflows/test_linters.yaml
@@ -10,7 +10,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.9"]
+        python-version: ["3.10"]
         os: [ubuntu-latest]
 
     steps:

diff --git a/.gitignore b/.gitignore
@@ -154,6 +154,7 @@ logs
 .DS_Store
 .devcontainer/devcontainer.json
 3.9
+3.10
 
 # PyCharm
 .idea/

diff --git a/Dockerfile b/Dockerfile
@@ -3,12 +3,12 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 # Ensure no installs try to launch interactive screen
 ARG DEBIAN_FRONTEND=noninteractive
 
-# Update packages and install python3.9 and other dependencies
+# Update packages and install python3.10 and other dependencies
 RUN apt-get update -y && \
     apt-get install -y software-properties-common git && \
     add-apt-repository -y ppa:deadsnakes/ppa && \
-    apt-get install -y python3.9 python3.9-dev python3-pip python3.9-venv && \
-    update-alternatives --install /usr/bin/python python /usr/bin/python3.9 10 && \
+    apt-get install -y python3.10 python3.10-dev python3-pip python3.10-venv && \
+    update-alternatives --install /usr/bin/python python /usr/bin/python3.10 10 && \
     python -m venv stoix && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*

diff --git a/README.md b/README.md
@@ -6,7 +6,7 @@
 
 <div align="center">
 <a href="https://www.python.org/doc/versions/">
-      <img src="https://img.shields.io/badge/python-3.9-blue" alt="Python Versions">
+      <img src="https://img.shields.io/badge/python-3.10-blue" alt="Python Versions">
 </a>
 <a  href="https://github.com/instadeepai/Mava/blob/main/LICENSE">
     <img src="https://img.shields.io/badge/License-Apache%202.0-orange.svg" alt="License" />
@@ -65,7 +65,7 @@ cd stoix
 pip install -e .
 ```
 
-We have tested `Stoix` on Python 3.9. Note that because the installation of JAX differs depending on your hardware accelerator,
+We have tested `Stoix` on Python 3.10. Note that because the installation of JAX differs depending on your hardware accelerator,
 we advise users to explicitly install the correct JAX version (see the [official installation guide](https://github.com/google/jax#installation)). For more in-depth installation guides including Docker builds and virtual environments, please see our [detailed installation guide](docs/DETAILED_INSTALL.md).
 
 ## Quickstart ⚡

diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ line-length = 100
 profile = "black"
 
 [tool.mypy]
-python_version = 3.9
+python_version = 3.10
 namespace_packages = true
 incremental = false
 cache_dir = ""

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -8,6 +8,7 @@ huggingface_hub
 hydra-core==1.3.2
 jax
 jaxlib
+jaxmarl
 jumanji @ git+https://github.com/instadeepai/jumanji.git@main
 neptune
 numpy

diff --git a/setup.py b/setup.py
@@ -30,7 +30,7 @@ def _get_version() -> str:
     long_description_content_type="text/markdown",
     keywords="single-agent reinforcement-learning python jax",
     packages=setuptools.find_packages(),
-    python_requires=">=3.9",
+    python_requires=">=3.10",
     install_requires=_parse_requirements("requirements/requirements.txt"),
     extras_require={
         "dev": _parse_requirements("requirements/requirements-dev.txt"),
@@ -42,7 +42,7 @@ def _get_version() -> str:
         "Intended Audience :: Science/Research",
         "Intended Audience :: Developers",
         "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
         "Topic :: Software Development :: Libraries :: Python Modules",
         "License :: OSI Approved :: Apache Software License",