Skip to content

Commit c40b4b2

Browse files
committed
build: Added pytest to test and mx to opt optional dependencies. Also minor ordering change
Signed-off-by: Brandon Groth <[email protected]>
1 parent 0bc05f5 commit c40b4b2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,15 @@ dependencies = [
4141
]
4242

4343
[project.optional-dependencies]
44-
dev = ["pre-commit>=3.0.4,<5.0"]
4544
fp8 = ["llmcompressor"]
4645
gptq = ["Cython", "gptqmodel>=1.7.3"]
4746
mx = ["microxcaling>=1.1"]
48-
test = ["pillow"]
47+
opt = ["fms-model-optimizer[fp8, gptq, mx]"]
4948
torchvision = ["torchvision>=0.17"]
50-
visualize = ["matplotlib", "graphviz", "pygraphviz"]
5149
flash-attn = ["flash-attn>=2.5.3,<3.0"]
52-
opt = ["fms-model-optimizer[fp8, gptq]"]
50+
visualize = ["matplotlib", "graphviz", "pygraphviz"]
51+
dev = ["pre-commit>=3.0.4,<5.0"]
52+
test = ["pytest", "pillow"]
5353

5454
[project.urls]
5555
homepage = "https://github.com/foundation-model-stack/fms-model-optimizer"

0 commit comments

Comments
 (0)