[tool.poetry]
name = "mantel"
version = "0.1.0"
description = ""
authors = ["Harry Stuart <hj.stuart0003@gmail.com>"]
readme = "README.md"
packages = [{ include = "models", from = "src" }, { include = "tests" }]


[tool.poetry.dependencies]
python = "^3.12"
numpy = "^1.26.4"
jax = "^0.4.28"
returns = "^0.22.0"
matplotlib = "^3.9.0"
jaxlib = "^0.4.28"
chex = "^0.1.86"
hypothesis = "^6.103.1"
mlflow = "^2.13.2"
pytest = "^8.2.2"
seaborn = "^0.13.2"
pandas = "^2.2.2"


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