diff --git a/setup.py b/setup.py
index 859091c..439e5e8 100644
--- a/setup.py
+++ b/setup.py
@@ -73,6 +73,7 @@ def read_requirements():
     'pandas',
     'psutil',
     'pytest',
+    'hypothesis',
     'PyYAML',
     'torch',
     'torchvision',