agilerl 2.4.3.dev0__tar.gz → 2.5.0.dev0__tar.gz

{agilerl-2.4.3.dev0 → agilerl-2.5.0.dev0}/.coveragerc

@@ -2,6 +2,7 @@
 concurrency = multiprocessing
 parallel = true
 sigterm = true
+omit = agilerl/networks/distributions_experimental.py
 
 [report]
 exclude_lines =

agilerl-2.5.0.dev0/.github/workflows/python-app.yml

@@ -0,0 +1,56 @@
+---
+name: Tests
+
+on:
+    push:
+        branches: [main, nightly]
+        paths:
+            - agilerl/**
+    pull_request:
+        branches: [main, nightly]
+        paths:
+            - agilerl/**
+
+permissions:
+    contents: read
+
+jobs:
+    build:
+        runs-on: gha-runner-scale-set
+        strategy:
+            fail-fast: false
+            max-parallel: 4
+            matrix:
+                python-version: ['3.10', '3.11', '3.12', '3.13']
+
+        container:
+            image: pytorch/pytorch:2.7.1-cuda12.6-cudnn9-devel
+            options: --user root
+
+        # Workspace (/__w) is ~1GB with little free space; root (/) has plenty. Put cache and venv on /.
+        env:
+            UV_CACHE_DIR: /tmp/uv-cache
+            UV_PROJECT_ENVIRONMENT: /tmp/agilerl-venv
+
+        steps:
+            - uses: actions/checkout@v4
+            - uses: astral-sh/setup-uv@v7
+              with:
+                  enable-cache: true
+                  python-version: ${{ matrix.python-version }}
+
+            - name: Install dependencies
+              run: |
+                  uv sync --locked --all-groups --extra all
+                  echo "/tmp/agilerl-venv/bin" >> $GITHUB_PATH
+                  uv cache clean
+                  rm -rf /tmp/uv-cache
+
+            - name: Test with pytest
+              run: |
+                  uv run pytest --exitfirst --cov=agilerl --cov-report=xml
+
+            - name: Upload coverage reports to Codecov
+              uses: codecov/codecov-action@v3
+              env:
+                  CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}

{agilerl-2.4.3.dev0 → agilerl-2.5.0.dev0}/.gitignore

@@ -106,10 +106,8 @@ celerybeat.pid
 # Environments
 .env
 .vscode/
-.venv
-# env/
+.venv*/
 venv/
-# ENV/
 env.bak/
 venv.bak/
 

{agilerl-2.4.3.dev0 → agilerl-2.5.0.dev0}/.pre-commit-config.yaml

@@ -23,10 +23,6 @@ repos:
           - id: debug-statements
           - id: mixed-line-ending
             args: [--fix=lf]
-    - repo: https://github.com/psf/black-pre-commit-mirror
-      rev: 26.1.0
-      hooks:
-          - id: black
     - repo: https://github.com/codespell-project/codespell
       rev: v2.4.1
       hooks:
@@ -35,24 +31,20 @@ repos:
                 - --skip=*.css,*.js,*.map,*.scss,*.svg
                 - --ignore-words-list=magent,pres,roate
     - repo: https://github.com/astral-sh/ruff-pre-commit
-      rev: v0.14.14
-      hooks:
-          - id: ruff-check
-            args:
-                - --select=I
-                - --fix
-                - --ignore=E501,F401
-    - repo: https://github.com/asottile/pyupgrade
-      rev: v3.21.2
+      rev: v0.15.1
       hooks:
-          - id: pyupgrade
-            args: [--py38-plus]
+          - id: ruff
+            name: Ruff Linter
+            args: [--fix]
+          - id: ruff-format
+            name: Ruff Formatter
+
     - repo: https://github.com/jumanjihouse/pre-commit-hook-yamlfmt
       rev: 0.2.3
       hooks:
           - id: yamlfmt
     - repo: https://github.com/astral-sh/uv-pre-commit
       # uv version.
-      rev: 0.9.28
+      rev: 0.10.2
       hooks:
           - id: uv-lock

{agilerl-2.4.3.dev0 → agilerl-2.5.0.dev0}/PKG-INFO

@@ -1,23 +1,23 @@
 Metadata-Version: 2.4
 Name: agilerl
-Version: 2.4.3.dev0
+Version: 2.5.0.dev0
 Summary: AgileRL is a deep reinforcement learning library focused on improving RL development through RLOps.
 Author-email: Nick Ustaran-Anderegg <dev@agilerl.com>
 License-Expression: Apache-2.0
 License-File: LICENSE
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.14,>=3.10
 Requires-Dist: accelerate~=1.7.0
 Requires-Dist: dill~=0.3.7
 Requires-Dist: fastrand~=1.3.0
 Requires-Dist: flatten-dict~=0.4.2
 Requires-Dist: google-cloud-storage~=2.5.0
 Requires-Dist: gymnasium~=1.0.0
-Requires-Dist: h5py~=3.8.0
+Requires-Dist: h5py~=3.15.0
 Requires-Dist: hydra-core~=1.3.2
 Requires-Dist: jax[cpu]~=0.4.31
 Requires-Dist: matplotlib<3.10,~=3.9.4
 Requires-Dist: minari[all]==0.5.2
-Requires-Dist: numpy~=1.26.4
+Requires-Dist: numpy<3.0,>=2.0.0
 Requires-Dist: omegaconf~=2.3.0
 Requires-Dist: packaging>=20.0
 Requires-Dist: pandas~=2.2.3
@@ -26,14 +26,12 @@ Requires-Dist: pre-commit~=3.8.0
 Requires-Dist: pygame~=2.6.0
 Requires-Dist: pymunk~=6.2.0
 Requires-Dist: redis~=4.4.4
-Requires-Dist: scipy~=1.12.0
 Requires-Dist: supersuit~=3.9.0
 Requires-Dist: tensordict~=0.8
 Requires-Dist: termcolor~=1.1.0
 Requires-Dist: torch==2.7.1
-Requires-Dist: tqdm~=4.66.4
-Requires-Dist: ucimlrepo~=0.0.3
-Requires-Dist: wandb~=0.17.6
+Requires-Dist: tqdm>=4.66.4
+Requires-Dist: wandb~=0.18.0
 Provides-Extra: all
 Requires-Dist: datasets==4.4.1; extra == 'all'
 Requires-Dist: deepspeed~=0.17.1; extra == 'all'
@@ -57,6 +55,7 @@ Description-Content-Type: text/markdown
 
 [![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
 [![Documentation Status](https://readthedocs.org/projects/agilerl/badge/?version=latest)](https://docs.agilerl.com/en/latest/?badge=latest)
+[![Tests](https://github.com/AgileRL/AgileRL/actions/workflows/python-app.yml/badge.svg)](https://github.com/AgileRL/AgileRL/actions/workflows/python-app.yml)
 [![Downloads](https://static.pepy.tech/badge/agilerl)](https://pypi.python.org/pypi/agilerl/)
 [![Discord](https://dcbadge.limes.pink/api/server/https://discord.gg/eB8HyTA2ux?style=flat)](https://discord.gg/eB8HyTA2ux)
 [![Arena](./.github/badges/arena-github-badge.svg)](https://arena.agilerl.com)

{agilerl-2.4.3.dev0 → agilerl-2.5.0.dev0}/README.md

@@ -7,6 +7,7 @@
 
 [![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
 [![Documentation Status](https://readthedocs.org/projects/agilerl/badge/?version=latest)](https://docs.agilerl.com/en/latest/?badge=latest)
+[![Tests](https://github.com/AgileRL/AgileRL/actions/workflows/python-app.yml/badge.svg)](https://github.com/AgileRL/AgileRL/actions/workflows/python-app.yml)
 [![Downloads](https://static.pepy.tech/badge/agilerl)](https://pypi.python.org/pypi/agilerl/)
 [![Discord](https://dcbadge.limes.pink/api/server/https://discord.gg/eB8HyTA2ux?style=flat)](https://discord.gg/eB8HyTA2ux)
 [![Arena](./.github/badges/arena-github-badge.svg)](https://arena.agilerl.com)