diff --git a/setup.py b/setup.py
index 3db1e16..4d6fb1d 100644
--- a/setup.py
+++ b/setup.py
@@ -8,5 +8,5 @@ setup(
     # author_email='author@gmail.com',
     # description='Description of my package',
     packages=['.'],
-    install_requires=[],
+    install_requires=['wandb'],
 )