diff --git a/setup.py b/setup.py
index fb06d97d5c..f39283e73c 100644
--- a/setup.py
+++ b/setup.py
@@ -26,6 +26,7 @@
     "papermill",
     "pykeops",
     "torchvision",
+    "pymoo",
 ]
 
 # Check for python version