diff --git a/setup.py b/setup.py
index c3a77d7..d260f04 100644
--- a/setup.py
+++ b/setup.py
@@ -15,6 +15,7 @@
         "matplotlib",
         "numpy>=1.23",
         "orion",
+        "hydra-orion-sweeper",
         "pyyaml>=5.3",
         "pytest",
         "pytorch_lightning==1.6.4",