pyproject.toml

[project]
name = "fair-bads"
version = "0.1.0"
description = ""
authors = [
    {name = "Your Name",email = "you@example.com"}
]
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "torch (>=2.6.0,<3.0.0)",
    "scikit-learn (>=1.6.1,<2.0.0)",
    "pot (>=0.9.5,<0.10.0)",
    "lightning (>=2.5.1.post0,<3.0.0)",
    "torchvision (>=0.22.0,<0.23.0)",
    "wandb (>=0.19.10,<0.20.0)",
    "gdown (>=5.2.0,<6.0.0)",
    "ftfy (>=6.3.1,<7.0.0)",
    "regex (>=2024.11.6,<2025.0.0)",
    "tqdm (>=4.67.1,<5.0.0)",
    "clip @ git+https://github.com/openai/CLIP.git"
]


[build-system]
requires = ["poetry-core>=2.0.0,<3.0.0"]
build-backend = "poetry.core.masonry.api"


[[tool.poetry.source]]
name = "torch-gpu"
url = "https://download.pytorch.org/whl/cu126"
priority = "explicit"

[tool.poetry.dependencies]
torch = {source = "torch-gpu"}