[project]
name = "ml-lab-mcp"
version = "0.1.0"
description = "MCP server for ML model training, fine-tuning, and experimentation"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "PolyForm-Noncommercial-1.0.0" }
authors = [{ name = "ML Lab Contributors" }]
keywords = ["mcp", "machine-learning", "fine-tuning", "llm", "training"]
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"License :: Other/Proprietary License",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
]
dependencies = [
"mcp>=1.0.0",
"httpx>=0.27.0",
"pydantic>=2.0.0",
"pydantic-settings>=2.0.0",
"cryptography>=41.0.0",
"aiosqlite>=0.19.0",
"rich>=13.0.0",
"typer>=0.9.0",
"pyyaml>=6.0.0",
]
[project.optional-dependencies]
dev = [
"pytest>=7.0.0",
"pytest-asyncio>=0.21.0",
"ruff>=0.1.0",
"mypy>=1.0.0",
]
training = [
"torch>=2.0.0",
"transformers>=4.36.0",
"datasets>=2.14.0",
"peft>=0.7.0",
"bitsandbytes>=0.41.0",
"accelerate>=0.25.0",
"trl>=0.7.0",
]
cloud = [
"modal>=0.55.0",
"paramiko>=3.0.0",
"boto3>=1.34.0",
"google-cloud-aiplatform>=1.38.0",
]
[project.scripts]
ml-lab = "ml_lab.cli:main"
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.hatch.build.targets.wheel]
packages = ["src/ml_lab"]
[tool.ruff]
line-length = 100
target-version = "py310"
[tool.ruff.lint]
select = ["E", "F", "I", "N", "W", "UP"]
[tool.mypy]
python_version = "3.10"
strict = true
[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]
