pyproject.toml

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "autogluon.assistant"
version = "0.1.1"
description = "ML Assistant for Competitive Machine Learning"
authors = [
    {name = "AutoGluon Community"}
]
readme = "README.md"
requires-python = ">=3.8,<3.12"
dependencies = [
    "autogluon.tabular[all]>=1.1,<2.0",
    "importlib-resources>=6.4.5,<7.0",
    "langchain>=0.3.3,<0.4",
    "langchain_openai>=0.2.2,<0.3",
    "langchain_aws>=0.2.2,<0.3",
    "pydantic>=2.9.2,<3.0",
    "hydra-core>=1.3",
    # TODO: remove matplotlib later, as it is not used in AGA
    # but needed for seamless pip install of autogluon.tabular[all]
    "matplotlib>=3.9.2",
    "typer>=0.12.5",
    "rich>=13.8.1",
    "s3fs>=2024.9.0",
    "joblib>=1.4.2",
    "python-calamine",
    "tenacity>=8.2.2,<10.0",
    "pandas>=2.2",
    "streamlit>=1.37,<1.41",
    "streamlit-aggrid>=1.0.2,<1.1",
    "streamlit-extras>=0.4,<0.6",
    "psutil>=5.9.8",
]

[project.optional-dependencies]
dev = [
    "black>=24",
    "GitRepo>=1",
    "pytest>=7",
    "isort",
    "ruff==0.4",
]

[project.urls]
"Homepage" = "https://github.com/autogluon/autogluon-assistant"

[project.scripts]
autogluon-assistant = "autogluon.assistant:main"
aga = "autogluon.assistant:main"

[tool.setuptools.packages.find]
where = ["src"]
include = ["autogluon.assistant*"]
namespaces = true

[tool.setuptools.package-data]
"autogluon.assistant" = [
    "configs/*.yaml",
    "ui/**/*",
    "ui/.streamlit/config.toml"  # Explicitly include the Streamlit config file
]

[tool.pytest.ini_options]
testpaths = "tests"
addopts = "--strict-markers"
xfail_strict = true

[tool.black]
line-length = 119
target-version = ['py38', 'py39', 'py310', 'py311']

[tool.isort]
known_first_party = "autogluon.assistant"
known_third_party = [
    "autogluon",
    "joblib",
    "langchain",
    "numpy",
    "omegaconf",
    "openai",
    "pandas",
    "psutil",
    "pydantic",
    "pytest",
    "requests",
    "scipy",
    "sklearn",
    "tqdm",
]
line_length = 119
profile = "black"

[tool.codespell]
skip = '.git,*.pdf,*.svg,*.ipynb,*.csv,'

[tool.ruff]
lint.ignore = [
    "E501",  # Line too long
    "E731",  # Do not assign a `lambda` expression, use a `def`
    "E722",  # Do not use bare `except`
]
extend-exclude = ["__init__.py"]

[tool.mypy]
ignore_missing_imports = true