ci

.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 torchvision av
+python -m pip install flake8 pytest pytest-cov hydra-core tqdm torch torch_geometric torchvision "av<14"
 
 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 torchvision av
+python -m pip install torch torchvision "av<14"
 # 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

@@ -43,7 +43,7 @@ You may want to install the following rendering and logging tools
 
 .. code-block:: console
 
-   pip install wandb moviepy torchvision av
+   pip install wandb moviepy torchvision "av<14"
 
 Install environments
 --------------------

setup.py

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