ml-inspector-mcp 0.1.0__tar.gz

ml_inspector_mcp-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,30 @@
+name: Publish to PyPI
+
+on:
+  push:
+    tags:
+      - "v*"
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      contents: read
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Build package
+        run: uv build
+
+      - name: Publish to PyPI
+        run: uv publish --token ${{ secrets.PYPI_API_TOKEN }}

ml_inspector_mcp-0.1.0/.github/workflows/test.yml

@@ -0,0 +1,31 @@
+name: Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Install dependencies
+        run: uv sync --extra sklearn-onnx --extra explain
+
+      - name: Generate demo model files
+        run: uv run python examples/train_demo_model.py
+
+      - name: Run tests
+        run: uv run python -m pytest tests/ -v

ml_inspector_mcp-0.1.0/.gitignore

@@ -0,0 +1,41 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+*.pyd
+*.egg
+*.egg-info/
+dist/
+build/
+.eggs/
+*.whl
+
+# uv / virtual envs
+.venv/
+venv/
+.python-version
+
+# Testing & coverage
+.pytest_cache/
+.coverage
+coverage.xml
+htmlcov/
+
+# Mypy
+.mypy_cache/
+
+# Ruff
+.ruff_cache/
+
+# IDEs
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Distribution
+MANIFEST

ml_inspector_mcp-0.1.0/PKG-INFO

@@ -0,0 +1,170 @@
+Metadata-Version: 2.5
+Name: ml-inspector-mcp
+Version: 0.1.0
+Summary: Framework-agnostic ML model analysis MCP server — evaluate, explain, and report on any trained model via Claude
+Project-URL: Homepage, https://github.com/YOUR_USERNAME/ml-inspector-mcp
+Project-URL: Repository, https://github.com/YOUR_USERNAME/ml-inspector-mcp
+License: MIT
+Keywords: claude,explainability,machine-learning,mcp,mlops,model-evaluation,shap
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.11
+Requires-Dist: anyio
+Requires-Dist: fastapi
+Requires-Dist: httpx
+Requires-Dist: jinja2
+Requires-Dist: joblib
+Requires-Dist: matplotlib
+Requires-Dist: mcp<2.0.0,>=1.0.0
+Requires-Dist: numpy
+Requires-Dist: onnx>=1.14.0
+Requires-Dist: onnxruntime>=1.16.0
+Requires-Dist: packaging
+Requires-Dist: pandas
+Requires-Dist: pillow
+Requires-Dist: pydantic>=2.0
+Requires-Dist: scikit-learn
+Requires-Dist: scipy
+Requires-Dist: seaborn
+Requires-Dist: uvicorn
+Provides-Extra: dev
+Requires-Dist: mypy>=1.10.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest-cov>=5.0.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Provides-Extra: drift
+Requires-Dist: evidently; extra == 'drift'
+Provides-Extra: explain
+Requires-Dist: shap; extra == 'explain'
+Provides-Extra: full
+Requires-Dist: anthropic; extra == 'full'
+Requires-Dist: evidently; extra == 'full'
+Requires-Dist: plotly; extra == 'full'
+Requires-Dist: shap; extra == 'full'
+Requires-Dist: skl2onnx; extra == 'full'
+Requires-Dist: tensorflow>=2.13; extra == 'full'
+Requires-Dist: tf2onnx>=1.15; extra == 'full'
+Requires-Dist: torch; extra == 'full'
+Requires-Dist: torchvision; extra == 'full'
+Requires-Dist: weasyprint; extra == 'full'
+Provides-Extra: pytorch
+Requires-Dist: torch; extra == 'pytorch'
+Requires-Dist: torchvision; extra == 'pytorch'
+Provides-Extra: reports
+Requires-Dist: anthropic; extra == 'reports'
+Requires-Dist: weasyprint; extra == 'reports'
+Provides-Extra: sklearn-onnx
+Requires-Dist: skl2onnx; extra == 'sklearn-onnx'
+Provides-Extra: tensorflow
+Requires-Dist: tensorflow>=2.13; extra == 'tensorflow'
+Requires-Dist: tf2onnx>=1.15; extra == 'tensorflow'
+Description-Content-Type: text/markdown
+
+# ml-inspector-mcp
+
+[![PyPI](https://img.shields.io/pypi/v/ml-inspector-mcp)](https://pypi.org/project/ml-inspector-mcp/)
+[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
+Framework-agnostic ML model analysis MCP server. Drop in any trained model
+and test data — Claude evaluates it, explains predictions, detects drift,
+and generates PDF reports via natural language.
+
+## Installation
+
+```bash
+pip install ml-inspector-mcp                                          # minimal
+pip install "ml-inspector-mcp[full]"                                  # everything
+pip install "ml-inspector-mcp[sklearn-onnx,explain,reports]"          # common combo
+```
+
+## Claude Desktop setup
+
+Add to `~/Library/Application Support/Claude/claude_desktop_config.json` (Mac):
+
+```json
+{
+  "mcpServers": {
+    "ml-inspector": {
+      "command": "ml-inspector",
+      "env": {
+        "ANTHROPIC_API_KEY": "your-key-here",
+        "MLFLOW_TRACKING_URI": "http://localhost:5000"
+      }
+    }
+  }
+}
+```
+
+## Quick start
+
+```bash
+python examples/train_demo_model.py
+```
+
+Then in Claude Desktop:
+> "Load the demo model from examples/demo_model.onnx"
+> "Load test data from examples/demo_test.csv"
+> "Evaluate the model and tell me how it's performing"
+> "Explain what drove the prediction for sample 5"
+> "Generate a PDF evaluation report"
+
+## Model compatibility
+
+| Format | Framework | Install |
+|--------|-----------|---------|
+| `.onnx` | Any | Always works — recommended |
+| `.pkl` / `.joblib` | scikit-learn | `pip install "ml-inspector-mcp[sklearn-onnx]"` |
+| `.h5` / `.keras` | TensorFlow/Keras | `pip install "ml-inspector-mcp[tensorflow]"` |
+| `.pt` / `.pth` | PyTorch (full model only) | `pip install "ml-inspector-mcp[pytorch]"` |
+
+### Version mismatch fix
+
+If you get version errors loading a `.pkl` or `.pt` file, export to ONNX first:
+
+```bash
+# scikit-learn — use the convert_to_onnx tool after loading, or:
+python -c "
+import joblib
+from skl2onnx import convert_sklearn
+from skl2onnx.common.data_types import FloatTensorType
+model = joblib.load('model.pkl')
+onnx_model = convert_sklearn(model, initial_types=[('input', FloatTensorType([None, N_FEATURES]))])
+open('model.onnx', 'wb').write(onnx_model.SerializeToString())
+"
+
+# PyTorch
+torch.onnx.export(model, dummy_input, "model.onnx", opset_version=17)
+
+# TensorFlow / Keras
+python -m tf2onnx.convert --keras model.h5 --output model.onnx
+```
+
+## All tools
+
+| Tool | Description |
+|------|-------------|
+| `load_model` | Load any model file (`.pkl`, `.h5`, `.pt`, `.onnx`) — auto-detects framework |
+| `get_model_info` | Info about the currently loaded model |
+| `convert_to_onnx` | Convert loaded model to ONNX format |
+| `list_supported_formats` | Show all supported formats and install instructions |
+| `load_test_data` | Load a CSV as test dataset |
+| `evaluate_model` | Full evaluation — accuracy, F1, AUC, confusion matrix, per-class metrics |
+| `find_worst_predictions` | Find samples the model struggled most with |
+| `evaluate_by_slice` | Evaluate on a data subset (e.g. by group or label) |
+| `threshold_analysis` | Sweep decision threshold — precision/recall/F1/FPR trade-offs |
+| `explain_prediction` | SHAP explanation for a single sample |
+| `global_feature_importance` | Mean absolute SHAP values across all samples |
+| `plot_shap_summary` | SHAP beeswarm summary plot saved as PNG |
+| `data_quality_report` | Null counts, class imbalance, outliers, data type warnings |
+| `detect_drift` | Statistical drift detection between two datasets (Evidently) |
+| `plot_confusion_matrix` | Confusion matrix heatmap (raw + normalized) saved as PNG |
+| `plot_roc_curve` | ROC curve with per-class AUC scores saved as PNG |
+| `generate_report` | Full PDF / HTML / Markdown report with metrics, charts, and optional AI narrative |
+
+## License
+
+MIT

ml_inspector_mcp-0.1.0/README.md

@@ -0,0 +1,105 @@
+# ml-inspector-mcp
+
+[![PyPI](https://img.shields.io/pypi/v/ml-inspector-mcp)](https://pypi.org/project/ml-inspector-mcp/)
+[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
+Framework-agnostic ML model analysis MCP server. Drop in any trained model
+and test data — Claude evaluates it, explains predictions, detects drift,
+and generates PDF reports via natural language.
+
+## Installation
+
+```bash
+pip install ml-inspector-mcp                                          # minimal
+pip install "ml-inspector-mcp[full]"                                  # everything
+pip install "ml-inspector-mcp[sklearn-onnx,explain,reports]"          # common combo
+```
+
+## Claude Desktop setup
+
+Add to `~/Library/Application Support/Claude/claude_desktop_config.json` (Mac):
+
+```json
+{
+  "mcpServers": {
+    "ml-inspector": {
+      "command": "ml-inspector",
+      "env": {
+        "ANTHROPIC_API_KEY": "your-key-here",
+        "MLFLOW_TRACKING_URI": "http://localhost:5000"
+      }
+    }
+  }
+}
+```
+
+## Quick start
+
+```bash
+python examples/train_demo_model.py
+```
+
+Then in Claude Desktop:
+> "Load the demo model from examples/demo_model.onnx"
+> "Load test data from examples/demo_test.csv"
+> "Evaluate the model and tell me how it's performing"
+> "Explain what drove the prediction for sample 5"
+> "Generate a PDF evaluation report"
+
+## Model compatibility
+
+| Format | Framework | Install |
+|--------|-----------|---------|
+| `.onnx` | Any | Always works — recommended |
+| `.pkl` / `.joblib` | scikit-learn | `pip install "ml-inspector-mcp[sklearn-onnx]"` |
+| `.h5` / `.keras` | TensorFlow/Keras | `pip install "ml-inspector-mcp[tensorflow]"` |
+| `.pt` / `.pth` | PyTorch (full model only) | `pip install "ml-inspector-mcp[pytorch]"` |
+
+### Version mismatch fix
+
+If you get version errors loading a `.pkl` or `.pt` file, export to ONNX first:
+
+```bash
+# scikit-learn — use the convert_to_onnx tool after loading, or:
+python -c "
+import joblib
+from skl2onnx import convert_sklearn
+from skl2onnx.common.data_types import FloatTensorType
+model = joblib.load('model.pkl')
+onnx_model = convert_sklearn(model, initial_types=[('input', FloatTensorType([None, N_FEATURES]))])
+open('model.onnx', 'wb').write(onnx_model.SerializeToString())
+"
+
+# PyTorch
+torch.onnx.export(model, dummy_input, "model.onnx", opset_version=17)
+
+# TensorFlow / Keras
+python -m tf2onnx.convert --keras model.h5 --output model.onnx
+```
+
+## All tools
+
+| Tool | Description |
+|------|-------------|
+| `load_model` | Load any model file (`.pkl`, `.h5`, `.pt`, `.onnx`) — auto-detects framework |
+| `get_model_info` | Info about the currently loaded model |
+| `convert_to_onnx` | Convert loaded model to ONNX format |
+| `list_supported_formats` | Show all supported formats and install instructions |
+| `load_test_data` | Load a CSV as test dataset |
+| `evaluate_model` | Full evaluation — accuracy, F1, AUC, confusion matrix, per-class metrics |
+| `find_worst_predictions` | Find samples the model struggled most with |
+| `evaluate_by_slice` | Evaluate on a data subset (e.g. by group or label) |
+| `threshold_analysis` | Sweep decision threshold — precision/recall/F1/FPR trade-offs |
+| `explain_prediction` | SHAP explanation for a single sample |
+| `global_feature_importance` | Mean absolute SHAP values across all samples |
+| `plot_shap_summary` | SHAP beeswarm summary plot saved as PNG |
+| `data_quality_report` | Null counts, class imbalance, outliers, data type warnings |
+| `detect_drift` | Statistical drift detection between two datasets (Evidently) |
+| `plot_confusion_matrix` | Confusion matrix heatmap (raw + normalized) saved as PNG |
+| `plot_roc_curve` | ROC curve with per-class AUC scores saved as PNG |
+| `generate_report` | Full PDF / HTML / Markdown report with metrics, charts, and optional AI narrative |
+
+## License
+
+MIT

ml_inspector_mcp-0.1.0/examples/demo_test.csv

@@ -0,0 +1,46 @@
+sepal_length,sepal_width,petal_length,petal_width,label
+6.1,2.8,4.7,1.2,versicolor
+5.7,3.8,1.7,0.3,setosa
+7.7,2.6,6.9,2.3,virginica
+6.0,2.9,4.5,1.5,versicolor
+6.8,2.8,4.8,1.4,versicolor
+5.4,3.4,1.5,0.4,setosa
+5.6,2.9,3.6,1.3,versicolor
+6.9,3.1,5.1,2.3,virginica
+6.2,2.2,4.5,1.5,versicolor
+5.8,2.7,3.9,1.2,versicolor
+6.5,3.2,5.1,2.0,virginica
+4.8,3.0,1.4,0.1,setosa
+5.5,3.5,1.3,0.2,setosa
+4.9,3.1,1.5,0.1,setosa
+5.1,3.8,1.5,0.3,setosa
+6.3,3.3,4.7,1.6,versicolor
+6.5,3.0,5.8,2.2,virginica
+5.6,2.5,3.9,1.1,versicolor
+5.7,2.8,4.5,1.3,versicolor
+6.4,2.8,5.6,2.2,virginica
+4.7,3.2,1.6,0.2,setosa
+6.1,3.0,4.9,1.8,virginica
+5.0,3.4,1.6,0.4,setosa
+6.4,2.8,5.6,2.1,virginica
+7.9,3.8,6.4,2.0,virginica
+6.7,3.0,5.2,2.3,virginica
+6.7,2.5,5.8,1.8,virginica
+6.8,3.2,5.9,2.3,virginica
+4.8,3.0,1.4,0.3,setosa
+4.8,3.1,1.6,0.2,setosa
+4.6,3.6,1.0,0.2,setosa
+5.7,4.4,1.5,0.4,setosa
+6.7,3.1,4.4,1.4,versicolor
+4.8,3.4,1.6,0.2,setosa
+4.4,3.2,1.3,0.2,setosa
+6.3,2.5,5.0,1.9,virginica
+6.4,3.2,4.5,1.5,versicolor
+5.2,3.5,1.5,0.2,setosa
+5.0,3.6,1.4,0.2,setosa
+5.2,4.1,1.5,0.1,setosa
+5.8,2.7,5.1,1.9,virginica
+6.0,3.4,4.5,1.6,versicolor
+6.7,3.1,4.7,1.5,versicolor
+5.4,3.9,1.3,0.4,setosa
+5.4,3.7,1.5,0.2,setosa

ml_inspector_mcp-0.1.0/examples/demo_train.csv

@@ -0,0 +1,106 @@
+sepal_length,sepal_width,petal_length,petal_width,label
+5.5,2.4,3.7,1.0,versicolor
+6.3,2.8,5.1,1.5,virginica
+6.4,3.1,5.5,1.8,virginica
+6.6,3.0,4.4,1.4,versicolor
+7.2,3.6,6.1,2.5,virginica
+5.7,2.9,4.2,1.3,versicolor
+7.6,3.0,6.6,2.1,virginica
+5.6,3.0,4.5,1.5,versicolor
+5.1,3.5,1.4,0.2,setosa
+7.7,2.8,6.7,2.0,virginica
+5.8,2.7,4.1,1.0,versicolor
+5.2,3.4,1.4,0.2,setosa
+5.0,3.5,1.3,0.3,setosa
+5.1,3.8,1.9,0.4,setosa
+5.0,2.0,3.5,1.0,versicolor
+6.3,2.7,4.9,1.8,virginica
+4.8,3.4,1.9,0.2,setosa
+5.0,3.0,1.6,0.2,setosa
+5.1,3.3,1.7,0.5,setosa
+5.6,2.7,4.2,1.3,versicolor
+5.1,3.4,1.5,0.2,setosa
+5.7,3.0,4.2,1.2,versicolor
+7.7,3.8,6.7,2.2,virginica
+4.6,3.2,1.4,0.2,setosa
+6.2,2.9,4.3,1.3,versicolor
+5.7,2.5,5.0,2.0,virginica
+5.5,4.2,1.4,0.2,setosa
+6.0,3.0,4.8,1.8,virginica
+5.8,2.7,5.1,1.9,virginica
+6.0,2.2,4.0,1.0,versicolor
+5.4,3.0,4.5,1.5,versicolor
+6.2,3.4,5.4,2.3,virginica
+5.5,2.3,4.0,1.3,versicolor
+5.4,3.9,1.7,0.4,setosa
+5.0,2.3,3.3,1.0,versicolor
+6.4,2.7,5.3,1.9,virginica
+5.0,3.3,1.4,0.2,setosa
+5.0,3.2,1.2,0.2,setosa
+5.5,2.4,3.8,1.1,versicolor
+6.7,3.0,5.0,1.7,versicolor
+4.9,3.1,1.5,0.2,setosa
+5.8,2.8,5.1,2.4,virginica
+5.0,3.4,1.5,0.2,setosa
+5.0,3.5,1.6,0.6,setosa
+5.9,3.2,4.8,1.8,versicolor
+5.1,2.5,3.0,1.1,versicolor
+6.9,3.2,5.7,2.3,virginica
+6.0,2.7,5.1,1.6,versicolor
+6.1,2.6,5.6,1.4,virginica
+7.7,3.0,6.1,2.3,virginica
+5.5,2.5,4.0,1.3,versicolor
+4.4,2.9,1.4,0.2,setosa
+4.3,3.0,1.1,0.1,setosa
+6.0,2.2,5.0,1.5,virginica
+7.2,3.2,6.0,1.8,virginica
+4.6,3.1,1.5,0.2,setosa
+5.1,3.5,1.4,0.3,setosa
+4.4,3.0,1.3,0.2,setosa
+6.3,2.5,4.9,1.5,versicolor
+6.3,3.4,5.6,2.4,virginica
+4.6,3.4,1.4,0.3,setosa
+6.8,3.0,5.5,2.1,virginica
+6.3,3.3,6.0,2.5,virginica
+4.7,3.2,1.3,0.2,setosa
+6.1,2.9,4.7,1.4,versicolor
+6.5,2.8,4.6,1.5,versicolor
+6.2,2.8,4.8,1.8,virginica
+7.0,3.2,4.7,1.4,versicolor
+6.4,3.2,5.3,2.3,virginica
+5.1,3.8,1.6,0.2,setosa
+6.9,3.1,5.4,2.1,virginica
+5.9,3.0,4.2,1.5,versicolor
+6.5,3.0,5.2,2.0,virginica
+5.7,2.6,3.5,1.0,versicolor
+5.2,2.7,3.9,1.4,versicolor
+6.1,3.0,4.6,1.4,versicolor
+4.5,2.3,1.3,0.3,setosa
+6.6,2.9,4.6,1.3,versicolor
+5.5,2.6,4.4,1.2,versicolor
+5.3,3.7,1.5,0.2,setosa
+5.6,3.0,4.1,1.3,versicolor
+7.3,2.9,6.3,1.8,virginica
+6.7,3.3,5.7,2.1,virginica
+5.1,3.7,1.5,0.4,setosa
+4.9,2.4,3.3,1.0,versicolor
+6.7,3.3,5.7,2.5,virginica
+7.2,3.0,5.8,1.6,virginica
+4.9,3.6,1.4,0.1,setosa
+6.7,3.1,5.6,2.4,virginica
+4.9,3.0,1.4,0.2,setosa
+6.9,3.1,4.9,1.5,versicolor
+7.4,2.8,6.1,1.9,virginica
+6.3,2.9,5.6,1.8,virginica
+5.7,2.8,4.1,1.3,versicolor
+6.5,3.0,5.5,1.8,virginica
+6.3,2.3,4.4,1.3,versicolor
+6.4,2.9,4.3,1.3,versicolor
+5.6,2.8,4.9,2.0,virginica
+5.9,3.0,5.1,1.8,virginica
+5.4,3.4,1.7,0.2,setosa
+6.1,2.8,4.0,1.3,versicolor
+4.9,2.5,4.5,1.7,virginica
+5.8,4.0,1.2,0.2,setosa
+5.8,2.6,4.0,1.2,versicolor
+7.1,3.0,5.9,2.1,virginica

ml_inspector_mcp-0.1.0/examples/train_demo_model.py

@@ -0,0 +1,49 @@
+"""Run this once to generate demo model files for testing ml-inspector-mcp."""
+import pandas as pd
+import numpy as np
+from sklearn.datasets import load_iris
+from sklearn.ensemble import RandomForestClassifier
+from sklearn.model_selection import train_test_split
+from pathlib import Path
+import joblib
+
+Path("examples").mkdir(exist_ok=True)
+
+iris = load_iris()
+X, y = iris.data, iris.target
+class_names = list(iris.target_names)  # ['setosa', 'versicolor', 'virginica']
+
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
+
+model = RandomForestClassifier(n_estimators=100, random_state=42)
+model.fit(X_train, y_train)
+
+# Save as sklearn pkl
+joblib.dump(model, "examples/demo_model.pkl")
+print("Saved examples/demo_model.pkl")
+
+# Save as ONNX (recommended)
+try:
+    from skl2onnx import convert_sklearn
+    from skl2onnx.common.data_types import FloatTensorType
+    onnx_model = convert_sklearn(model, initial_types=[("input", FloatTensorType([None, 4]))])
+    with open("examples/demo_model.onnx", "wb") as f:
+        f.write(onnx_model.SerializeToString())
+    print("Saved examples/demo_model.onnx (recommended)")
+except ImportError:
+    print("skl2onnx not installed, skipping ONNX export. pip install skl2onnx")
+
+# Save train/test CSVs
+feature_cols = ["sepal_length", "sepal_width", "petal_length", "petal_width"]
+train_df = pd.DataFrame(X_train, columns=feature_cols)
+train_df["label"] = [class_names[i] for i in y_train]
+train_df.to_csv("examples/demo_train.csv", index=False)
+
+test_df = pd.DataFrame(X_test, columns=feature_cols)
+test_df["label"] = [class_names[i] for i in y_test]
+test_df.to_csv("examples/demo_test.csv", index=False)
+
+print("Saved examples/demo_train.csv and examples/demo_test.csv")
+print("\nDemo files ready. Load in Claude Desktop:")
+print("  load_model('examples/demo_model.onnx', 'classification')")
+print("  load_test_data('examples/demo_test.csv', 'label')")

ml_inspector_mcp-0.1.0/pyproject.toml

@@ -0,0 +1,98 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "ml-inspector-mcp"
+version = "0.1.0"
+description = "Framework-agnostic ML model analysis MCP server — evaluate, explain, and report on any trained model via Claude"
+readme = "README.md"
+requires-python = ">=3.11"
+license = {text = "MIT"}
+keywords = ["mcp", "machine-learning", "model-evaluation", "explainability", "mlops", "shap", "claude"]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Topic :: Scientific/Engineering :: Artificial Intelligence",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+    "mcp>=1.0.0,<2.0.0",
+    "fastapi",
+    "uvicorn",
+    "pydantic>=2.0",
+    "numpy",
+    "pandas",
+    "scikit-learn",
+    "onnx>=1.14.0",
+    "onnxruntime>=1.16.0",
+    "joblib",
+    "matplotlib",
+    "seaborn",
+    "pillow",
+    "scipy",
+    "jinja2",
+    "packaging",
+    "anyio",
+    "httpx",
+]
+
+[project.urls]
+Homepage = "https://github.com/YOUR_USERNAME/ml-inspector-mcp"
+Repository = "https://github.com/YOUR_USERNAME/ml-inspector-mcp"
+
+[project.scripts]
+ml-inspector = "ml_inspector.server:main"
+
+[project.optional-dependencies]
+tensorflow  = ["tensorflow>=2.13", "tf2onnx>=1.15"]
+pytorch     = ["torch", "torchvision"]
+sklearn-onnx = ["skl2onnx"]
+explain     = ["shap"]
+drift       = ["evidently"]
+reports     = ["weasyprint", "anthropic"]
+full = [
+    "tensorflow>=2.13", "tf2onnx>=1.15",
+    "torch", "torchvision",
+    "skl2onnx", "shap",
+    "evidently", "weasyprint", "anthropic", "plotly",
+]
+
+# Development & CI
+dev = [
+    "pytest>=8.0.0",
+    "pytest-asyncio>=0.23.0",
+    "pytest-cov>=5.0.0",
+    "ruff>=0.4.0",
+    "mypy>=1.10.0",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/ml_inspector"]
+
+# ── Ruff (linter + formatter) ──────────────────────────────────────────────
+[tool.ruff]
+target-version = "py311"
+line-length = 88
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "UP", "B", "SIM"]
+ignore = ["E501"]
+
+[tool.ruff.lint.isort]
+known-first-party = ["ml_inspector"]
+
+# ── Mypy ──────────────────────────────────────────────────────────────────
+[tool.mypy]
+python_version = "3.11"
+strict = true
+ignore_missing_imports = true
+
+# ── Pytest ────────────────────────────────────────────────────────────────
+[tool.pytest.ini_options]
+testpaths    = ["tests"]
+asyncio_mode = "auto"
+
+[tool.coverage.run]
+source = ["src/ml_inspector"]
+omit   = ["tests/*"]