[tool.poetry]
name = "streamlit-audience"
version = "0.1.0"
description = "Demo application to showcase artist similarity models"
authors = ["Rain <rain.bomberg@theorchard.com>"]

[tool.poetry.dependencies]
python = "3.8.13"
streamlit = "^1.12.2"
pandas = "^1.4.3"
numpy = "^1.23.2"
faiss-cpu = "^1.7.2"
annoy = "^1.17.1"
mypy = "^0.971"
pydantic = "^1.10.0"
pandas-stubs = "^1.4.3"
boto3 = "^1.24.64"
boto3-stubs = "^1.24.64"
python-dotenv = "^0.21.0"

[tool.poetry.dev-dependencies]
isort = "^5.10.1"
black = "^22.6.0"
flake8 = "^5.0.4"

[tool.black]
line-length = 88
target-version = ['py38']
include = '\.pyi?$'

[tool.isort]
profile = "black"
combine_as_imports = true
known_tests = "tests"
sections = ["FUTURE", "STDLIB", "THIRDPARTY", "FIRSTPARTY", "TESTS", "LOCALFOLDER"]

[tool.mypy]
python_version = "3.8"
strict = true
show_error_codes = true

[[tool.mypy.overrides]]
module = [
    "faiss.*",
    "pyarrow.*",
    "pyarrow.compute.*",
    "matplotlib.*",
    "nltk.stem.*",
    "boto3.*",
    "pandas.*",
    "s3fs.*",
    "pandarallel.*",
    "audience.functions.*",
]
ignore_missing_imports = true

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