requirements.txt

# --------- pytorch --------- #
# temporary fix for cross-platform torch install with cuda compatibility
torch>=2.0.0; sys_platform != "win32" # "darwin" and "linux"
torchvision>=0.15.0; sys_platform != "win32"
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
# 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
anndata==0.10.6
networkx==3.2.1
numpy==1.26.4
numba==0.59.1
omegaconf==2.3.0
packaging==24.0
pandas==2.2.2
pertpy==0.7.0
pytest==8.1.1
rich==13.7.1
rootutils==1.0.7
scanpy==1.10.0
scipy==1.13.1
setuptools==68.2.0
tqdm==4.66.2
gdown==5.2.0
wandb===0.18.3