[Deps] Make pytest an optional dependency (#610)

Pytest is currently included in wheel dependencies, but it is
needed only to run the unit tests. Move it to optional dependencies
under the "test" group.

Signed-off-by: Will Daly <[email protected]>

Author: wedaly

pyproject.toml

@@ -20,12 +20,12 @@ dependencies = [
     "transformers>=4.53.0",
     "datasets>=3.3.0",
     "einops",
-    "pytest",
 ]
 
 [project.optional-dependencies]
 conv1d = ["causal-conv1d>=1.4.0"]
 benchmark = ["matplotlib"]
+test = ["pytest"]
 
 [project.urls]
 Homepage = "https://github.com/fla-org/flash-linear-attention"

setup.py

@@ -42,11 +42,11 @@ def get_package_version():
         'torch>=2.5',
         'transformers>=4.53.0',
         'datasets>=3.3.0',
-        'einops',
-        'pytest'
+        'einops'
     ],
     extras_require={
         'conv1d': ['causal-conv1d>=1.4.0'],
         'benchmark': ['matplotlib'],
+        'test': ['pytest'],
     }
 )