build: Add pillow optional dependency for testing

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

pyproject.toml

@@ -45,7 +45,8 @@ dev = ["pre-commit>=3.0.4,<5.0"]
 fp8 = ["llmcompressor"]
 gptq = ["Cython", "gptqmodel>=1.7.3"]
 mx = ["microxcaling>=1.1"]
-vision = ["torchvision>=0.17"]
+test = ["pillow"]
+torchvision = ["torchvision>=0.17"]
 visualize = ["matplotlib", "graphviz", "pygraphviz"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
 opt = ["fms-model-optimizer[fp8, gptq]"]

tox.ini

@@ -6,6 +6,7 @@ minversion = 4.4
 description = run tests (unit, unitcov)
 extras =
     dev
+    test
 package = wheel
 wheel_build_env = pkg
 deps =