[Version] 1.2.0 (#80)

benchmarl/__init__.py

@@ -5,7 +5,7 @@
 #
 
 
-__version__ = "1.1.1"
+__version__ = "1.2.0"
 
 import importlib
 

docs/source/conf.py

@@ -34,8 +34,8 @@
     "python": ("https://docs.python.org/3/", None),
     "sphinx": ("https://www.sphinx-doc.org/en/master/", None),
     "torch": ("https://pytorch.org/docs/master", None),
-    "torchrl": ("https://pytorch.org/rl", None),
-    "tensordict": ("https://pytorch.org/tensordict", None),
+    "torchrl": ("https://pytorch.org/rl/stable/", None),
+    "tensordict": ("https://pytorch.org/tensordict/stable", None),
 }
 intersphinx_disabled_domains = ["std"]
 

docs/source/usage/installation.rst

@@ -17,7 +17,7 @@ For more details, or for installing nightly versions, see the
 Install BenchMARL
 -----------------
 
-You can just install it from github
+You can just install it from PyPi
 
 .. code-block:: console
 
@@ -48,7 +48,7 @@ You may want to install the following rendering and logging tools
 Install environments
 --------------------
 
-All enviornment dependencies are optional in BenchMARL and can be installed separately.
+All environment dependencies are optional in BenchMARL and can be installed separately.
 
 VMAS
 ^^^^

setup.py

@@ -46,7 +46,7 @@ def get_version():
     url="https://github.com/facebookresearch/BenchMARL",
     author="Matteo Bettini",
     author_email="[email protected]",
-    install_requires=["torchrl>=0.3.1", "tqdm", "hydra-core"],
+    install_requires=["torchrl==0.4.0", "tqdm", "hydra-core"],
     extras_require={
         "vmas": ["vmas>=1.3.4"],
         "pettingzoo": ["pettingzoo[all]>=1.24.3"],

test/test_meltingpot.py

@@ -5,9 +5,7 @@
 #
 
 
-import packaging
 import pytest
-import torchrl
 
 from benchmarl.algorithms import (
     algorithm_config_registry,
@@ -35,10 +33,6 @@ def _get_unique_envs(names):
 
 
 @pytest.mark.skipif(not _has_meltingpot, reason="Meltingpot not found")
-@pytest.mark.skipif(
-    packaging.version.parse(torchrl.__version__).base_version <= "0.3.1",
-    reason="TorchRL <= 0.3.1 does nto support meltingpot",
-)
 class TestMeltingPot:
     @pytest.mark.parametrize("algo_config", algorithm_config_registry.values())
     @pytest.mark.parametrize("task", [MeltingPotTask.COMMONS_HARVEST__OPEN])

test/test_models.py

@@ -6,10 +6,8 @@
 
 from typing import List
 
-import packaging
 import pytest
 import torch
-import torchrl
 
 from benchmarl.hydra_config import load_model_config_from_hydra
 from benchmarl.models import model_config_registry
@@ -79,11 +77,6 @@ def test_models_forward_shape(
         pytest.skip()  # this combination should never happen
     if ("gnn" in model_name) and centralised:
         pytest.skip("gnn model is always decentralized")
-    if (
-        packaging.version.parse(torchrl.__version__).base_version <= "0.3.1"
-        and "cnn" in model_name
-    ):
-        pytest.skip("TorchRL <= 0.3.1 does not support MultiAgentCNN")
 
     torch.manual_seed(0)