A code snippet for plotting a job from hydra-zen

hydra_zen_plot_landscape.py

def plot_jobs(*jobs, fn):
    import numpy as np
    import matplotlib.pyplot as plt
    
    (jobs,) = jobs
    x, y = np.arange(-2, 2, 0.1), np.arange(-2, 2, 0.1)
    X, Y = np.meshgrid(x, y)
    V = parabaloid(X, Y)
    fig, ax = plt.subplots()

    ax.contour(X, Y, V.reshape(X.shape), levels=[0.01, 0.12, 0.3, 0.5], colors="black")
    for job in jobs[0]:
        title, label = job.hydra_cfg.hydra.overrides.task[0].split("=")
        ax.plot(
            *job.return_value.T,
            label=label,
        )

    ax.legend(title=title)