pyproject.toml

[tool.poetry]
name = "mlserver"
version = "1.4.0.dev3"
description = "MLServer"
authors = ["Seldon Technologies Ltd. <hello@seldon.io>"]
license = "Apache-2.0"
readme = "README.md"
classifiers = [
    "Operating System :: POSIX",
    "Operating System :: MacOS"
]
include =  ["mlserver/rest/openapi/*.json"]

[tool.poetry.scripts]
mlserver = 'mlserver.cli:main'

[tool.black]
exclude = '''
(
  mlserver\.egg-info
  | \.mypy_cache
  | \.git
  | \.tox
  | dist
  | venv
)
'''

[tool.mypy]
ignore_missing_imports = true
plugins = "pydantic.mypy"
exclude = [
  'mlserver\.egg-info',
  '\.mypy_cache',
  '\.git',
  '\.tox',
  'dist',
  'venv',
]

[tool.pytest.ini_options]
asyncio_mode = "auto"
addopts = "--import-mode=importlib"

[tool.poetry.dependencies]
python = "^3.8.1,<3.12"
click = "*"
fastapi = ">=0.88.0,!=0.89.0,<=0.97.0"
python-dotenv = "*"
grpcio = "*"
numpy = "*"
pandas = "*"
protobuf = "*"
uvicorn = "*"
starlette-exporter = "*"
py-grpc-prometheus = "*"
aiokafka = "*"
tritonclient = {version = "^2.24", extras = ["http"]}
aiofiles = "*"
orjson = "*"
uvloop = {version = "*", markers = "sys_platform != 'win32' and (sys_platform != 'cygwin' and platform_python_implementation != 'PyPy')"}

## The importlib-resources backport is required to use some
## functionality added in Python 3.10
## https://setuptools.pypa.io/en/latest/userguide/datafiles.html#accessing-data-files-at-runtime
importlib-resources = "^5.12.0"
opentelemetry-instrumentation-fastapi = "^0.39b0"
opentelemetry-sdk = "^1.18.0"
opentelemetry-exporter-otlp-proto-grpc = "^1.18.0"
opentelemetry-instrumentation-grpc = "^0.39b0"

[tool.poetry.group.dev.dependencies]
datamodel-code-generator = "0.21.0"
grpcio-tools = "1.48.1"
tritonclient = {version = "2.32.0", extras = ["http"]}
pytest = "7.4.0"
pytest-asyncio = "0.21.0"
pytest-mock = "3.11.1"
pytest-cases = "3.6.14"
tox = "4.6.3"
docker = "6.0.1"
aiohttp = "3.8.5"
aiohttp-retry = "2.8.3"
## Used for FastAPI Async testing
httpx = "0.24.1"
kafka-python = "2.0.2"
tenacity = "8.2.2"
pyyaml = "6.0.1"

## Latest version of conda-pack (0.7.0) is not on PyPi
## See https://github.com/conda/conda-pack/issues/219
conda-pack = {git = "https://github.com/conda/conda-pack", rev = "0.7.0"}
flake8 = "6.0.0"
flake8-black = "0.3.6"
mypy = "1.5.0"
mypy-protobuf = "3.1.0"
types-protobuf = "4.23.0.1"
types-orjson = "3.6.2"
types-aiofiles = "23.1.0.4"
types-requests = "2.28.11.5"
black = "23.3.0"
pip-licenses = "4.3.2"

[tool.poetry.group.docker.dependencies]
protobuf = "3.20.3"
tensorflow = "~2.12.0"

[tool.poetry.group.all-runtimes]
optional = true

[tool.poetry.group.all-runtimes.dependencies]
mlserver-sklearn = {path = "./runtimes/sklearn", develop = true}
mlserver-xgboost = {path = "./runtimes/xgboost", develop = true}
mlserver-lightgbm = {path = "./runtimes/lightgbm", develop = true}
mlserver-mlflow = {path = "./runtimes/mlflow", develop = true}
mlserver-huggingface = {path = "./runtimes/huggingface", develop = true}
mlserver-alibi-explain = {path = "./runtimes/alibi-explain", develop = true}
mlserver-alibi-detect = {path = "./runtimes/alibi-detect", develop = true}

[tool.poetry.group.all-runtimes-dev]
optional = true

[tool.poetry.group.all-runtimes-dev.dependencies]
## Dev dependencies from Alibi-Explain
requests-mock = "~1.10.0"
types-requests = "~2.28.11.5"

## Dev dependencies from Alibi-Detect
tensorflow = "~2.12.0"

## Dev dependencies from MLflow
torch = "2.0.0"
pytorch-lightning = "2.0.7"
torchmetrics = "0.11.4"
torchvision = "0.15.1"
mlflow = "2.5.0"

## Dev dependencies from HuggingFace
transformers = "~4.30.0"

[tool.poetry.group.docs]
optional = true

[tool.poetry.group.docs.dependencies]
Sphinx = "6.2.1"
sphinx_material = "0.0.35"
readthedocs-sphinx-search = "0.3.1"
myst-parser = "1.0.0"
sphinxcontrib-bibtex = "2.5.0"
autodoc_pydantic = "1.9.0"
sphinx-click = "4.4.0"
sphinx_design = "0.4.1"
sphinx-autobuild = "2021.3.14"

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