parallel_trends_plot
Plot outcome means over time for treatment and control groups to visually assess parallel trends assumption before causal analysis.
Instructions
Plot raw outcome means over time for treatment and control groups.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| y | Yes | Outcome variable. | |
| ax | No | Existing axes to plot on. | |
| ci | No | Show 95% confidence intervals (+/-1.96 SE of mean). | |
| id | No | Unit identifier (for panel data). | |
| agg | No | Aggregation function: 'mean' or 'median'. | mean |
| time | Yes | Time period variable. | |
| title | No | Plot title. | |
| treat | Yes | Treatment group indicator. Binary (0/1) for 2x2, or first-treatment-period for staggered (0 = never treated). | |
| colors | No | Colors for (treatment, control). Default: ('#E74C3C', '#2C3E50'). | |
| detail | No | Payload depth: 'minimal' (~150 tokens) for sub-step calls where only the point estimate is needed; 'standard' (~1K tokens) for diagnostics + coefficient table; 'agent' (~2K tokens, default) adds violations / next_steps / suggested_functions so the LLM can plan its next call without another round-trip. | agent |
| labels | No | Custom labels, e.g. ``{'treat': 'New Jersey', 'control': 'Pennsylvania'}``. | |
| figsize | No | Figure size. | |
| as_handle | No | If true, cache the fitted result on the server and return result_id + result_uri alongside the JSON payload so a subsequent tools/call can chain without re-running. | |
| data_path | Yes | Absolute path or URL to a data file. Supported: .csv / .tsv / .txt (delimited), .parquet / .pq, .feather / .arrow, .xlsx / .xls, .dta (Stata), .json / .jsonl. Schemes: file://, s3://, gs://, https://. | |
| result_id | No | Optional handle to a previously-fitted result (returned by an earlier call when as_handle=true). Tools that operate on a fitted object accept this in place of re-supplying data_path + columns. | |
| treat_time | No | Treatment onset time. Draws a vertical line if provided. | |
| data_columns | No | Optional column projection. Parquet/Feather/Stata loaders honour this for fast partial reads. | |
| data_sample_n | No | Optional uniform random subsample size (seed=0, deterministic) — useful on huge panels. |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
No arguments | |||