requirements.txt

# Core Libraries
numpy==1.26.4
matplotlib==3.7.5
toml==0.10.2
scikit-learn>=1.5.0
#scikit-learn==1.3.2
scipy==1.14.1
tensorflow==2.17.0

# Additional Libraries
shap==0.44.1
pycatch22==0.4.4
scikit-optimize==0.10.2
pyswarms==1.3.0
pytest==8.3.3
xlsxwriter==3.2.0
types-toml
types-setuptools
wheel

# downstream
sktime==0.35.0
statsmodels==0.14.4
prophet==1.1.6
plotly==5.24.1
darts<0.34.0


# PyTorch and Related Libraries
torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1
# neuralforecast==1.6.4


# PyTorch Geometric and Extensions (use prebuilt binaries)
#-f https://data.pyg.org/whl/torch-2.5.1.html
#torch-geometric==2.6.1
#torch-cluster==1.6.3

# Other Dependencies
ujson==5.10.0
ipdb==0.13.13
pandas==2.0.3
torchcde==0.2.5
tables==3.10.1
MissForest==4.2.3
lightgbm==4.6.0
xgboost==2.1.4


# extractor
tsfel==0.1.9
tsfresh==0.20.3

# grin
einops==0.8.0


# automl
ray==2.41.0
ray[tune]==2.41.0


fire==0.7.0
fastdtw==0.3.4


# dev
torch-geometric==2.6.1
pytorch-lightning==2.5.0
torchmetrics==1.6.1
lightning==2.5.0