Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ A plugin for Claude Code and other AI coding platforms providing [Agent Skills](
| Skill | Description |
|-------|-------------|
| [pymc-modeling](skills/pymc-modeling/) | Bayesian statistical modeling with PyMC v5+. Covers model specification, MCMC inference (nutpie, NumPyro), ArviZ diagnostics, hierarchical models, GLMs, GPs, BART, time series, and more. |
| [pymc-mlflow](skills/pymc-mlflow/) | MLflow integration for PyMC models. Covers experiment tracking, InferenceData artifact management, autologging with pymc_marketing.mlflow, metadata strategies, and production deployment patterns for MMM and CLV models. |
| [pymc-testing](skills/pymc-testing/) | Testing PyMC models with pytest. Covers mock sampling with `pymc.testing.mock_sample`, pytest fixtures, and the distinction between fast structure-only tests (mocking) and slow posterior inference tests. |
| [marimo-notebook](skills/marimo-notebook/) | Reactive Python notebooks with marimo. Covers CLI, UI components, layout, SQL integration, caching, state management, and wigglystuff widgets. |

Expand Down Expand Up @@ -72,6 +73,8 @@ python-analytics-skills/
│ ├── pymc-modeling/
│ │ ├── SKILL.md # Main skill instructions
│ │ └── references/ # 12 detailed reference docs
│ ├── pymc-mlflow/
│ │ ├── SKILL.md # Main skill instructions
│ ├── pymc-testing/
│ │ ├── SKILL.md # Main skill instructions
│ │ └── references/
Expand All @@ -92,6 +95,7 @@ python-analytics-skills/

The plugin includes a `UserPromptSubmit` hook that suggests relevant skills when it detects keywords in your prompt:

- **PyMC + MLflow keywords**: mlflow, experiment tracking, artifact logging, model registry, deployment, mmm, clv, marketing mix model, customer lifetime value
- **PyMC keywords**: bayesian, pymc, mcmc, posterior, inference, arviz, prior, sampling, divergence, hierarchical model, gaussian process, bart, etc.
- **Marimo keywords**: marimo, reactive notebook, @app.cell, mo.ui, etc.

Expand Down
5 changes: 5 additions & 0 deletions skills.json
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,11 @@
"name": "pymc-testing",
"description": "Testing PyMC models with pytest",
"version": "1.0.0"
},
{
"name": "pymc-mlflow",
"description": "MLflow integration for PyMC models. Use when tracking experiments, logging artifacts (InferenceData, diagnostics, posterior predictive checks), comparing model runs, or preparing models for deployment.",
"version": "1.0.0"
}
]
}
Loading