Skip to content

Commit 7092543

Browse files
committed
add optional dependencies for time-series, cec and xgboost
1 parent 602e859 commit 7092543

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

pyproject.toml

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,14 +45,28 @@ viz = [
4545
ml = [
4646
"scikit-learn",
4747
]
48+
# CEC benchmark data fetching
49+
cec = [
50+
"pooch>=1.5.0",
51+
]
52+
# Time-series functions with sktime
53+
timeseries = [
54+
"surfaces[ml]",
55+
"sktime>=0.24.0",
56+
]
4857
# Image classification with CNNs (requires tensorflow)
4958
images = [
5059
"surfaces[ml]",
5160
"tensorflow>=2.10.0",
5261
]
62+
# XGBoost-based functions
63+
xgboost = [
64+
"surfaces[ml]",
65+
"xgboost>=1.7.0",
66+
]
5367
# All optional features
5468
full = [
55-
"surfaces[viz,ml,images]",
69+
"surfaces[viz,ml,cec,timeseries,images,xgboost]",
5670
]
5771
# Development
5872
dev = [

0 commit comments

Comments
 (0)