pyro-train/pyproject.toml at main · pyronear/pyro-train · 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
[project]
name = "pyro-train"
version = "0.3.0"
description = "Machine Learning Training pipeline for Wildfire Detection 🔥"
authors = [{ name = "Pyronear", email = "contact@pyronear.org" }]
requires-python = ">=3.11"
readme = "README.md"
dependencies = [
    "ultralytics==8.4.21",
    "pandas>=2.2.2,<3",
    "ncnn>=1.0.20250503",
    "pnnx>=1.0.20250503",
    "onnx>=1.18.0",
    "onnxruntime>=1.22.0",
    "onnxslim>=0.1.71",
    "tqdm>=4.66.0",
    "pyroengine @ git+https://github.com/pyronear/pyro-engine.git",
    "pyroclient @ git+https://github.com/pyronear/pyro-api.git#subdirectory=client",
    "fiftyone>=1.13.4",
    "huggingface-hub>=1.8.0",
]

[dependency-groups]
dev = [
    "dvc>=3.50.0,<4",
    "mlflow>=2.12.1,<3",
    "ipython>=8.23.0,<9",
    "mypy>=1.9.0,<2",
    "dvc-s3>=3.2.0,<4",
    "ruff>=0.12.7",
    "pre-commit>=4.2.0",
    "pytest>=8.3.5,<9",
]

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

[tool.hatch.metadata]
allow-direct-references = true