# --------- pytorch --------- #
torch==2.1.0
torchvision>=0.15.0
lightning>=2.0.0
torchmetrics>=0.11.4

# --------- hydra --------- #
hydra-core==1.3.2
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0

# --------- loggers --------- #
wandb==0.16.5
# neptune-client
# mlflow
# comet-ml
# aim>=3.16.2  # no lower than 3.16.2, see https://github.com/aimhubio/aim/issues/2550

# --------- others --------- #
rootutils       # standardizing the project root setup
pre-commit      # hooks for applying linters on commit
rich            # beautiful text formatting in terminal
pytest          # tests
# sh            # for running bash commands in some tests (linux/macos only)


# Our dependencies:
POT==0.9.3
dirichlet==0.9
einops==0.7.0
selene-sdk==0.4.4
biopython==1.83
pyBigWig==0.3.22
pyranges==0.0.129
cooler==0.9.3
cooltools==0.7.0
rdkit==2023.9.5
torch_geometric==2.5.3
torchdiffeq==0.2.3
schedulefree==1.2.5
pydantic==2.7.1
