chore(ml): uv (#16725)

* poetry to uv

* update ci

* remove caching

* add typeshed to dev

* no need for `--non-interactive`

* move backends to extras

* oopsie

* update ci

machine-learning/pyproject.toml

@@ -1,72 +1,77 @@
-[tool.poetry]
+[project]
 name = "machine-learning"
 version = "1.129.0"
 description = ""
-authors = ["Hau Tran <alex.tran1502@gmail.com>"]
+authors = [{ name = "Hau Tran", email = "alex.tran1502@gmail.com" }]
+requires-python = ">=3.10,<4.0"
 readme = "README.md"
-packages = [{include = "app"}]
+dependencies = [
+    "aiocache>=0.12.1,<1.0",
+    "fastapi>=0.95.2,<1.0",
+    "ftfy>=6.1.1",
+    "gunicorn>=21.1.0",
+    "huggingface-hub>=0.20.1,<1.0",
+    "insightface>=0.7.3,<1.0",
+    "opencv-python-headless>=4.7.0.72,<5.0",
+    "orjson>=3.9.5",
+    "pillow>=9.5.0,<11.0",
+    "pydantic>=2.0.0,<3",
+    "pydantic-settings>=2.5.2,<3",
+    "python-multipart>=0.0.6,<1.0",
+    "rich>=13.4.2",
+    "tokenizers>=0.15.0,<1.0",
+    "uvicorn[standard]>=0.22.0,<1.0",
+]
 
-[tool.poetry.dependencies]
-python = ">=3.10,<4.0"
-insightface = ">=0.7.3,<1.0"
-opencv-python-headless = ">=4.7.0.72,<5.0"
-pillow = ">=9.5.0,<11.0"
-fastapi = ">=0.95.2,<1.0"
-uvicorn = {extras = ["standard"], version = ">=0.22.0,<1.0"}
-pydantic = "^2.0.0"
-pydantic-settings = "^2.5.2"
-aiocache = ">=0.12.1,<1.0"
-rich = ">=13.4.2"
-ftfy = ">=6.1.1"
-python-multipart = ">=0.0.6,<1.0"
-orjson = ">=3.9.5"
-gunicorn = ">=21.1.0"
-huggingface-hub = ">=0.20.1,<1.0"
-tokenizers = ">=0.15.0,<1.0"
+[dependency-groups]
+test = [
+    "httpx>=0.24.1",
+    "pytest>=7.3.1",
+    "pytest-asyncio>=0.21.0",
+    "pytest-cov>=4.1.0",
+    "pytest-mock>=3.11.1",
+]
+types = [
+    "types-pyyaml>=6.0.12.20241230",
+    "types-requests>=2.32.0.20250306",
+    "types-setuptools>=75.8.2.20250305",
+    "types-simplejson>=3.20.0.20250218",
+    "types-ujson>=5.10.0.20240515",
+]
+lint = [
+    "black>=23.3.0",
+    "mypy>=1.3.0",
+    "ruff>=0.0.272",
+    { include-group = "types" },
+]
+dev = ["locust>=2.15.1", { include-group = "test" }, { include-group = "lint" }]
 
-[tool.poetry.group.dev.dependencies]
-mypy = ">=1.3.0"
-black = ">=23.3.0"
-pytest = ">=7.3.1"
-locust = ">=2.15.1"
-httpx = ">=0.24.1"
-pytest-asyncio = ">=0.21.0"
-pytest-cov = ">=4.1.0"
-ruff = ">=0.0.272"
-pytest-mock = ">=3.11.1"
+[project.optional-dependencies]
+cpu = ["onnxruntime>=1.15.0,<2"]
+cuda = ["onnxruntime-gpu>=1.17.0,<2"]
+openvino = ["onnxruntime-openvino>=1.17.1,<1.19.0"]
+armnn = ["onnxruntime>=1.15.0,<2"]
 
-[tool.poetry.group.cpu]
-optional = true
+[tool.uv]
+compile-bytecode = true
 
-[tool.poetry.group.cpu.dependencies]
-onnxruntime = "^1.15.0"
-
-[tool.poetry.group.cuda]
-optional = true
-
-[tool.poetry.group.cuda.dependencies]
-onnxruntime-gpu = {version = "^1.17.0", source = "cuda12"}
-
-[tool.poetry.group.openvino]
-optional = true
-
-[tool.poetry.group.openvino.dependencies]
-onnxruntime-openvino = ">=1.17.1,<1.19.0"
-
-[tool.poetry.group.armnn]
-optional = true
-
-[tool.poetry.group.armnn.dependencies]
-onnxruntime = "^1.15.0"
-
-[[tool.poetry.source]]
+[[tool.uv.index]]
 name = "cuda12"
 url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
-priority = "explicit"
+explicit = true
+
+[tool.uv.sources]
+onnxruntime-gpu = { index = "cuda12" }
+
+[tool.hatch.build.targets.sdist]
+include = ["app"]
+
+[tool.hatch.build.targets.wheel]
+include = ["app"]
 
 [build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
 [tool.mypy]
 python_version = "3.11"