Merge pull request #48 from fedebotu/main

Add CI tests, move to `pyproject.toml`

.github/workflows/tests.yml

@@ -0,0 +1,39 @@
+name: Tests
+on: [push, pull_request]
+
+jobs:
+  build:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: true
+      max-parallel: 15
+      matrix:
+        # os: [ubuntu-latest, macos-latest, windows-latest, macos-13-xlarge]
+        # For Apple Silicon: https://github.com/actions/runner-images/issues/8439
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
+        # Exclude Python 3.8 on Windows because of resolution issues
+        exclude:
+          - os: windows-latest
+            python-version: '3.8'
+    defaults:
+      run:
+        shell: bash
+    steps:
+    - name: Check out repository
+      uses: actions/checkout@v3
+
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v4
+      with:
+        python-version: ${{ matrix.python-version }}
+        cache: 'pip'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install pytest
+        pip install -e '.[dev]'
+
+    - name: Run pytest
+      run: pytest tests

jaxmarl/__init__.py

@@ -1,3 +1,4 @@
 from .registration import make, registered_envs
 
-__all__ = ["make", "registered_envs"]
+__all__ = ["make", "registered_envs"]
+__version__ = "0.0.2"

pyproject.toml

@@ -0,0 +1,41 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+include = ['jaxmarl*']
+
+[tool.setuptools.dynamic]
+version = {attr = "jaxmarl.__version__"}
+dependencies = {file = ["requirements/requirements.txt"]}
+optional-dependencies = {dev = { file = ["requirements/requirements-dev.txt"] }, qlearning = { file = ["requirements-qlearning.txt"] }}
+
+[project]
+name = "jaxmarl"
+readme = "README.md"
+description = "Multi-Agent Reinforcement Learning with JAX"
+authors = [
+    {name = "Foerster Lab for AI Research", email = "arutherford@robots.ox.ac.uk"},
+  ]
+dynamic = ["version", "dependencies", "optional-dependencies"]
+license = {file = "LICENSE"}
+requires-python = ">=3.8"
+classifiers = [
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Operating System :: OS Independent",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "License :: OSI Approved :: Apache Software License",
+]
+
+[project.urls]
+"Homepage" = "https://github.com/FLAIROx/JaxMARL"
+"Bug Tracker" = "https://github.com/FLAIROx/JaxMARL/issues"

requirements/requirements-dev.txt

@@ -1,3 +1,4 @@
 pytest
+pygame
 pettingzoo>=1.23.1
 tqdm>=4.65.0