add missing test requirement

bitsandbytes-foundation · Aug 21, 2024 · fe8b3bc · fe8b3bc
1 parent 7476f6b
commit fe8b3bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@ def has_ext_modules(self):
     install_requires=["torch", "numpy"],
     extras_require={
         "benchmark": ["pandas", "matplotlib"],
-        "test": ["scipy"],
+        "test": ["scipy", "lion_pytorch"],
     },
     long_description=read("README.md"),
     long_description_content_type="text/markdown",