[tool.poetry]
name = "africat"
version = "0.1.1"
description = "Categorise articles about Africa according to the AllAfrica_NewsStory dataset, based on categories from AllAfrica::Cats"
authors = ["Timothy Allen <tim@treehouse.org.za>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.11"
numpy = "^1.26.2"
pandas = "^2.1.3"
pandarallel = "^1.6.5"
torch = "^2.1.1"
torchtext = "^0.16.1"
torchdata = "^0.7.1"
tqdm = "^4.66.1"

[tool.poetry.scripts]
aa_create_database = "africat.aa_create_database:main"
categorise = "africat:categorise:main"

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