From fe8b3bc5e704ca5f0a11e72b1c479ac00d227a67 Mon Sep 17 00:00:00 2001
From: Titus von Koeller <9048635+Titus-von-Koeller@users.noreply.github.com>
Date: Wed, 21 Aug 2024 15:59:04 +0000
Subject: [PATCH] add missing test requirement

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 5eada03fd..6a92a9dce 100644
--- 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",