HyperTor/pyproject.toml at main · das-lab/HyperTor · 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
[tool.poetry]
name = "hypertor"
version = "1.5.51"
description = "Imbalanced darknet (Tor) traffic classification."
authors = ["__z__ <imzqqq@hotmail.com>"]
license = "AGPL-3.0-or-later"
readme = "README.md"
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Topic :: Communications :: Tor",
]
packages = [
    { include = "hypertor" },
]

[[tool.poetry.source]]
name = "tsinghua"
url = "https://pypi.tuna.tsinghua.edu.cn/simple/"
# priority = "primary"
priority = "supplemental"
[[tool.poetry.source]]
name = "aliyun"
url = "https://mirrors.aliyun.com/pypi/simple/"
priority = "supplemental"
[[tool.poetry.source]]
name = "nvidia"
url = "https://pypi.nvidia.com"
priority = "explicit"


[tool.poetry.dependencies]
python = ">=3.12,<3.15"
ipykernel = "^7.1.0"
pytest = "^9.0.2"
setuptools_rust = ">=1.12.0"
attrs = "^25.4.0"
numpy = "^2.4.1"
fastai = "^2.8.6"
scikit-learn = "^1.8.0"
scipy = "^1.17.1"
scipy-stubs = {extras = ["scipy"], version = "^1.17.1.0"}
pandas = "^3.0.0"
pandas-stubs = "^3.0.0.260204"
ipywidgets = "^8.1.7"
seaborn = "^0.13.2"
requests = "^2.32.5"
matplotlib = "^3.10.8"
tqdm = "^4.67.1"
opencv-python = "^4.10.0.84"
pyyaml = "^6.0.2"
tensorboard = "^2.20.0"
torch-tb-profiler = "^0.4.3"
pydantic = "^2.12.5"
pillow = "^12.1.0"
scapy = "^2.6.1"
netron = "^8.8.1"
absl-py = "^2.3.1"

torch = { version = "^2.9.1", optional = true }
torcheval = { version = "^0.0.7", optional = true }
# View model summaries in PyTorch!
torchinfo = { version = "^1.8.0", optional = true }
torchvision = { version = "^0.25.0", optional = true }
tensorrt = { version = "^10.12.0", optional = true }
onnx = { version = "^1.20.0", optional = true }
onnx-graphsurgeon = { version = "^0.5.8", optional = true }
onnxruntime-tools = { version = "^1.7.0", optional = true }
onnxruntime = { version = "^1.23.2", optional = true }
# `pip install --extra-index-url https://pypi.nvidia.com/ tensorrt-cu13-libs`
tensorrt-cu13-libs = { version = "^10.14.1", source = "nvidia", optional = true }
nvidia-cusolver-cu12 = { version = "^11.7.3.90", source = "nvidia", optional = true }
nvidia-cudnn-cu12 = { version = "^9.10.2.21", source = "nvidia", optional = true }

# `poetry install --extras gpu`
[tool.poetry.extras]
gpu = [
    "torch",
    "torcheval",
    "torchinfo",
    "torchvision",
    "tensorrt",
    "onnx",
    "onnx-graphsurgeon",
    "onnxconverter-common",
    "onnxruntime-tools",
    "onnxruntime",
]

# `poetry install --with llm`
[tool.poetry.group.llm.dependencies]
safetensors = "^0.7.0"
transformers = "^4.57.3"


[tool.poetry.group.dev.dependencies]
setuptools = "80.0.0" # `Tensorboard requires this version or below`
wheel = "^0.46.3"
cryptography = "^46.0.5"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"