u-moec-tiny/pyproject.toml at main · johnjg75dev/u-moec-tiny · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
[project]
name = "u-moec-tiny"
version = "0.1.0"
description = "Universal Mixture-of-Experts lossless neural compressor (Tier 1 scaffold)"
readme = "README.md"
requires-python = ">=3.10"
authors = [{name = "U-MoEC Authors"}]
license = {text = "Apache-2.0"}

dependencies = [
    # Core ML / compression
    "torch>=2.3",
    "numpy>=1.24",
    "pyyaml>=6.0",
    "einops>=0.7",
    "tqdm>=4.66",
    "datasets>=2.18",
    "pyarrow>=14",
    "textual>=0.50.0",
    # GUI / server
    "dash>=2.14",
    "dash-bootstrap-components>=1.5",
    "plotly>=5.18",
    "pandas>=2.0",
    "fastapi>=0.108",
    "uvicorn[standard]>=0.25",
    "aiohttp>=3.9",
    "rich>=13.0",
    "psutil>=5.9",
]

[project.optional-dependencies]
dev = ["pytest", "ruff", "mypy"]
train = ["wandb", "liger-kernel", "flash-attn"]
mamba = ["mamba-ssm"]
synthetic = ["Pillow"]

[project.scripts]
umoec-train       = "bin.train:main"
umoec-compress    = "bin.compress:main"
umoec-decompress  = "bin.decompress:main"
umoec-gui         = "gui.app:main"
umoec-server      = "gui.server:main"
umoec-tui         = "tui.app:main"
umoec             = "bin.umoec:main"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["umoec"]

[tool.ruff]
line-length = 100
target-version = "py310"

[tool.ruff.lint]
select = ["E", "F", "W", "I", "B", "UP"]
ignore = ["E501"]

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-q"