Skip to content

Commit 8608a8e

Browse files
committed
Adding optional dependencies
ml: For machine learning tools fit: For fitting tools
1 parent 62434da commit 8608a8e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pyproject.toml

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,19 +8,18 @@ dependencies= [
88
'scipy',
99
'awkward',
1010
'tqdm',
11-
'joblib',
12-
'tensorflow==2.18.0', # Use latest allowed by zfit
1311
'numpy',
14-
'scikit-learn',
1512
'toml',
16-
'zfit',
1713
'numpy',
1814
'matplotlib',
1915
'mplhep',
2016
'hist[plot]',
2117
'pandas']
2218

2319
[project.optional-dependencies]
20+
# Use latest tensorflow allowed by zfit
21+
fit = ['zfit','tensorflow==2.18.0']
22+
ml = ['scikit-learn', 'joblib']
2423
dev = ['pytest']
2524

2625
[tools.setuptools.packages.find]

0 commit comments

Comments
 (0)