diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh
index a75ba321..e20187a3 100644
--- a/.github/unittest/install_dependencies.sh
+++ b/.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
+python -m pip install flake8 pytest pytest-cov hydra-core tqdm torch_geometric
 
 if [ -f requirements.txt ]; then pip install -r requirements.txt; fi