particleflow/pyproject.toml at main · jpata/particleflow · 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "particleflow"
version = "3.1.0"
authors = [
  { name = "Joosep Pata", email = "joosep.pata@cern.ch" },
  { name = "Farouk Mokhtar" },
  { name = "Eric Wulff" },
  { name = "Javier Duarte" }
]
description = "Machine Learning for Particle Flow Reconstruction"
readme = "README.md"
requires-python = "==3.11.*"
license = { text = "Apache Software License" }
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: OS Independent",
    "Topic :: Scientific/Engineering :: Physics",
]
dependencies = [
    "array-record~=0.8.3",
    "autopep8",
    "awkward",
    "boost_histogram",
    "click",
    "comet-ml",
    "fastjet",
    "fsspec",
    "jupyter",
    "jupyter-book",
    "keras",
    "keras-tuner",
    "matplotlib",
    "mlcroissant",
    "mplhep",
    "networkx",
    "notebook",
    "numba",
    "numpy",
    "onnx",
    "onnxscript",
    "pandas",
    "papermill",
    "plotly",
    "pre-commit",
    "protobuf~=5.28.3",
    "psutil",
    #needed for apache_beam compatibility
    "pyarrow~=18.1.0; sys_platform == 'linux'",
    #needed to prevent mutex lock error on OSX
    "pyarrow~=22.0.0; sys_platform == 'darwin'",
    "pydantic",
    "pytest",
    "ray[tune]",
    "scikit-learn",
    "scikit-optimize",
    "scipy",
    "seaborn",
    "setGPU",
    "snakemake",
    "tensorboard",
    "tensorflow~=2.20.0",
    "tensorflow-datasets~=4.9.9",
    "tensorflow-metadata~=1.17.3",
    "torch==2.9.0",
    "torchvision",
    "torch_runstats",
    "torch-scatter",
    "tqdm",
    "uproot",
    "umap-learn",
    "vector",
    "zenodo_get",
    "huggingface_hub",
    #required for dataset preparation, only on linux
    "apache_beam~=2.71.0; sys_platform == 'linux'",
    #required for litept
    "spconv-cu124; sys_platform == 'linux'",
    "addict",
    "timm",
    "colorhash",
    "flash-attn; sys_platform == 'linux'",
]

[project.scripts]
mlpf = "mlpf.pipeline:main"

[tool.uv.sources]
torch = [
  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
torchvision = [
  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
flash-attn = { url = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.9-cp311-cp311-linux_x86_64.whl" }
torch-scatter = { url = "https://data.pyg.org/whl/torch-2.9.0%2Bcu128/torch_scatter-2.1.2%2Bpt29cu128-cp311-cp311-linux_x86_64.whl" }

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true


[tool.uv.extra-build-dependencies]
torch-scatter = ["torch"]
#pointrope = ["torch"]

[tool.setuptools]
include-package-data = true

[tool.setuptools.packages.find]
exclude = ["tests*", "notebooks*", "images*", "docs*", "scripts*"]

[tool.setuptools.package-data]
"*" = ["*.yaml"]