[CI] Fix CI (#79)

* amend

* [Quality] Automatically get version

* amend

* amend

Author: matteobettini

.github/unittest/install_dependencies.sh

@@ -1,7 +1,7 @@
 
 
 python -m pip install --upgrade pip
-python -m pip install flake8 pytest pytest-cov hydra-core tqdm torch torch_geometric
+python -m pip install flake8 pytest pytest-cov hydra-core tqdm torch torch_geometric torchvision av
 
 if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 

.github/unittest/install_dependencies_nightly.sh

@@ -5,7 +5,7 @@ python -m pip install flake8 pytest pytest-cov hydra-core tqdm torch_geometric
 
 if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
-python -m pip install torch
+python -m pip install torch torchvision av
 # Not using nightly torch
 # python -m pip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu --force-reinstall
 

docs/source/usage/installation.rst

@@ -30,6 +30,21 @@ Or also clone it locally to access the configs and scripts
     git clone https://github.com/facebookresearch/BenchMARL.git
     pip install -e BenchMARL
 
+Install optional packages
+-------------------------
+
+By default, BenchMARL has only the core requirements.
+Here are some optional packages you may want to install.
+
+Logging
+^^^^^^^
+
+You may want to install the following rendering and logging tools
+
+.. code-block:: console
+
+   pip install wandb moviepy torchvision av
+
 Install environments
 --------------------
 

setup.py

@@ -52,6 +52,7 @@ def get_version():
         "pettingzoo": ["pettingzoo[all]>=1.24.3"],
         "meltingpot": ["dm-meltingpot"],
         "gnn": ["torch_geometric"],
+        "logging": ["moviepy", "wandb", "torchvision"],
     },
     packages=find_packages(),
     include_package_data=True,

test/test_models.py

@@ -170,6 +170,6 @@ def test_models_forward_shape(
         agent_group="agents",
         action_spec=None,
     )
-    input_td = input_spec.rand()
+    input_td = input_spec.expand(batch_size).rand()
     out_td = model(input_td)
     assert output_spec.expand(batch_size).is_in(out_td)