feat: update plot API to use flat functions from PyAutoFit PR #1174

[Jammy2211]

Overview

PyAutoFit PR PyAutoLabs/PyAutoFit#1174 replaced the class-based plotting API (NestPlotter, MCMCPlotter, MLEPlotter, SamplesPlotter) with simple module-level functions (e.g. aplt.corner_anesthetic(samples=s)). The autofit_workspace still uses the old class-based API in several scripts and has stale docstrings referencing the removed classes. This task updates all workspace scripts to use the new flat function API.

Plan

• Migrate 6 scripts that still instantiate old plotter classes to the new flat function API
• Update docstrings/comments in all 6 scripts/plot/*.py files that reference old class names
• Regenerate notebooks from updated scripts
• Verify no references to old plotter classes remain

Detailed implementation plan

Affected Repositories

• autofit_workspace (primary)

Work Classification

Workspace

Branch Survey

Repository	Current Branch	Dirty?
autofit_workspace	main	smoke_tests.txt modified

Suggested branch: feature/plot-api-update

Implementation Steps

1. scripts/cookbooks/samples.py — Replace plotter = aplt.MCMCPlotter(samples=result.samples); plotter.corner_cornerpy() with aplt.corner_cornerpy(samples=result.samples). Update docstring.

2. scripts/cookbooks/search.py — Replace plotter = aplt.MCMCPlotter(samples=samples); plotter.corner_cornerpy(...) with aplt.corner_cornerpy(samples=samples, ...). Update docstring.

3. scripts/overview/overview_1_the_basics.py — Replace plotter = aplt.NestPlotter(samples=result.samples); plotter.corner_anesthetic() with aplt.corner_anesthetic(samples=result.samples).

4. scripts/howtofit/chapter_1_introduction/tutorial_5_results_and_samples.py — Replace plotter = aplt.MCMCPlotter(samples=result.samples); plotter.corner_cornerpy() with aplt.corner_cornerpy(samples=result.samples). Update docstring.

5. scripts/howtofit/chapter_3_graphical_models/tutorial_1_individual_models.py — Replace plotter = aplt.NestPlotter(samples=samples); plotter.corner_cornerpy() with aplt.corner_cornerpy(samples=samples).

6. scripts/howtofit/chapter_3_graphical_models/tutorial_3_graphical_benefits.py — Replace plotter = aplt.NestPlotter(samples=result_list[0].samples); plotter.corner_cornerpy() with aplt.corner_cornerpy(samples=result_list[0].samples).

7. scripts/plot/*.py (6 files) — Update docstrings only (DynestyPlotter, NautilusPlotter, UltraNestPlotter, EmceePlotter, ZeusPlotter, PySwarmsPlotter).

8. Regenerate notebooks from scripts.

Key Files

• scripts/cookbooks/samples.py
• scripts/cookbooks/search.py
• scripts/overview/overview_1_the_basics.py
• scripts/howtofit/chapter_1_introduction/tutorial_5_results_and_samples.py
• scripts/howtofit/chapter_3_graphical_models/tutorial_1_individual_models.py
• scripts/howtofit/chapter_3_graphical_models/tutorial_3_graphical_benefits.py
• scripts/plot/DynestyPlotter.py
• scripts/plot/NautilusPlotter.py
• scripts/plot/UltraNestPlotter.py
• scripts/plot/EmceePlotter.py
• scripts/plot/ZeusPlotter.py
• scripts/plot/PySwarmsPlotter.py

Original Prompt

Click to expand starting prompt

The following issue implements a plot interface udpate PyAutoLabs/PyAutoFit#1174

This never got implemented in @autofit_workspace, so we just need to apply the API updates there.

Do a quick check locally that indeed the old API is still present.

[Jammy2211]

Smoke Test Results — 2026-04-05

Workspace	Passed	Failed	Total
autofit_workspace	7	0	7
autogalaxy_workspace	8	0	8
autolens_workspace	7	3	10
autolens_workspace_test	3	4	7

All autofit_workspace smoke tests pass. Failures in autolens workspaces are pre-existing and unrelated to the plot API update (JAX tracer errors, missing datasets, existing syntax errors, aplt.Output removal in autolens_workspace_test).

[Jammy2211]

Shipped — 2026-04-05

PRs

• Update plot API to flat functions (PyAutoFit PR #1174) #19

Summary

Updated all autofit_workspace scripts and docstrings to use the new flat function plot API from PyAutoFit PR #1174. Migrated 6 scripts from class-based plotter instantiation to direct module-level function calls, and updated docstrings in all 6 plot example scripts. All 7 autofit_workspace smoke tests pass.

[Jammy2211]

Shipped — 2026-04-05

PRs

• Update plot API to flat functions (PyAutoFit PR #1174) #19 (merged)

Summary

Updated all autofit_workspace scripts and docstrings to use the new flat function plot API from PyAutoFit PR #1174. Migrated 6 scripts from class-based plotter instantiation to direct module-level function calls, and updated docstrings in all 6 plot example scripts. All 7 autofit_workspace smoke tests pass. Remaining failures in autolens workspaces are pre-existing and being addressed separately.