pyproject.toml

[build-system]
requires = ["setuptools>=61.0.0", "wheel", "setuptools_scm[toml]>=6.2"]
build-backend = "setuptools.build_meta"

[project]
name = "ethnicolr2"
version = "0.1.1"
description = "Predict Race/Ethnicity Based on Sequence of Characters in the Name"
readme = "README.rst"
authors = [
    {name = "Rajashekar Chintalapati, Suriyan Laohaprapanon, Gaurav Sood", email = "rajshekar.ch@gmail.com, suriyant@gmail.com, gsood07@gmail.com"}
]
license = {text = "MIT"}
classifiers = [
    "Development Status :: 4 - Beta",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3.11",
    "Topic :: Scientific/Engineering :: Information Analysis",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Topic :: Utilities",
]
keywords = ["race", "ethnicity", "names"]
requires-python = ">=3.7"
dependencies = [
    "pandas>=1.3.0",
    "torch==2.2.0",
    "numpy>=1.20.0,<2.0.0",  # Pin to NumPy 1.x to avoid PyTorch compatibility issues
    "joblib==1.3.1",
    "tqdm==4.66.3",
    "scikit-learn==1.2.2",   # Use exact version that trained the models
]

[project.urls]
"Homepage" = "https://github.com/appeler/ethnicolr2"
"Repository" = "https://github.com/appeler/ethnicolr2"
"Bug Tracker" = "https://github.com/appeler/ethnicolr2/issues"

[project.optional-dependencies]
dev = ["check-manifest"]
test = ["coverage", "tox"]
docs = ["sphinx", "sphinx_rtd_theme"]
all = ["check-manifest", "coverage", "tox", "sphinx", "sphinx_rtd_theme"]

[project.scripts]
census_ln = "ethnicolr2.census_ln:main"
pred_fl_full_name = "ethnicolr2.pred_fl_full_name:main"
pred_fl_last_name = "ethnicolr2.pred_fl_last_name:main"
pred_census_last_name = "ethnicolr2.pred_census_last_name:main"
ethnicolr2_download_models = "ethnicolr2._setup_hooks:download_cli"

# Modern setuptools hooks (replaces setup.py custom commands)
[project.entry-points."setuptools.finalize_distribution_options"]
post-setup = "ethnicolr2._setup_hooks:register_commands"

[tool.setuptools]
packages = ["ethnicolr2"]
include-package-data = true

[tool.setuptools.package-data]
ethnicolr2 = [
    "data/census/census_2000.csv",
    "data/census/census_2010.csv",
    "data/census/readme.md",
    "data/census/*.pdf",
    "data/census/*.R",
    "models/*.joblib",
    "models/*.pt",
    "models/census/lstm/*.csv",
    "data/input*.csv",
    "examples/*.ipynb",
]

# Testing configuration
[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = "test_*.py"
addopts = "--verbose"

# Code quality and formatting tools
[tool.black]
line-length = 88
target-version = ["py37", "py38", "py39", "py310", "py311"]

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

[tool.flake8]
max-line-length = 88
extend-ignore = "E203"