pyproject.toml

[project]
name = "detoxify"
version = "0.5.2"
description = "A python library for detecting toxic comments"
readme = "README.md"
authors = [
    {name = "Unitary", email = "laura@unitary.ai"},
]
classifiers = [
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
]
requires-python = ">=3.9,<3.13"
dependencies = [
    "sentencepiece >= 0.1.94",
    "torch >=2",
    "transformers >= 3",
]

[tool.setuptools]
packages = ["detoxify"]

[project.urls]
Homepage = "https://github.com/unitaryai/detoxify"

[project.optional-dependencies]
dev = [
    "datasets >= 1.0.2",
    "pandas >= 1.1.2",
    "pytest",
    "pytorch-lightning>2",
    "scikit-learn >= 0.23.2",
    "tqdm",
    "pre-commit",
    "numpy<2"
]

[tool.black]
# https://github.com/psf/black
line-length = 120
exclude = "(.eggs|.git|.hg|.mypy_cache|.venv|_build|buck-out|build|dist)"

[tool.isort]
known_first_party = [
    "detoxify",
]
skip_glob = []
profile = "black"
line_length = 120

[tool.pytest.ini_options]
norecursedirs = [
    ".git",
    "dist",
    "build",
]
addopts = [
    "--strict-markers",
    "--doctest-modules",
    "--durations=0",
    "--color=yes",
]
filterwarnings = [
    "ignore:.*deprecated alias.*:DeprecationWarning:tensorboard.*tensorflow_stub*:",
    "ignore:.*deprecated alias.*:DeprecationWarning:tensorboard.*tensor_util*:",
    "ignore:.*deprecated alias.*:DeprecationWarning:pyarrow.*pandas_compat*:",
]

[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"