pyproject.toml

[tool.poetry]
name = "fair-kmeans"
version = "0.1.2"
description = "Fair K-Means produces a fair clustering assignment according to the fairness definition of Chierichetti et al. Each point has a binary color, and the goal is to assign the points to clusters such that the number of points with different colors in each cluster is the same and the cost of the clusters is minimized."
authors = ["Melanie Schmidt <mschmidt@hhu.de>", "Giulia Baldini <giulia.baldini@hhu.de>"]
readme = "README.md"
license = "LICENSE"
packages = [
    { include = "fair_kmeans" }
]

[tool.poetry.dependencies]
python = "^3.9"
scikit-learn = "^1.6.1"
numpy = "^1.26.4"


[tool.poetry.group.dev.dependencies]
black = {extras = ["jupyter"], version = "^24.4.2"}
pre-commit = "^2.14.0"
flake8 = "^7.0.0"
mypy = "^1.10.0"


[tool.poetry.group.test.dependencies]
pandas = "^2.2.2"

[tool.poetry.build]
script = "build_extension.py"
generate-setup-file = true

[build-system]
requires = ["poetry-core", "setuptools"]
build-backend = "poetry.core.masonry.api"

[virtualenvs]
create = true
in-project = true

[tool.isort]
profile = "black"
line_length = 88