DOC add example for unimport

[MarcoGorelli]

No description provided.

[MarcoGorelli]

this failed when I tried using on PyMC3's notebooks, need to figure out why

it seems to turn

# %%
fig, ax = plt.subplots(2, 2, figsize=(12, 12))
ax = ax.flatten()
contour_pot(pot1f, ax[0], 'pot1', );
contour_pot(pot2f, ax[1], 'pot2');
contour_pot(pot3f, ax[2], 'pot3');
contour_pot(pot4f, ax[3], 'pot4');
fig.tight_layout()

# %%
from pymc3.distributions.dist_math import bound


def cust_logp(z):
    #return bound(-pot1(z), z>-5, z<5)
    return -pot1(z)

with pm.Model() as pot1m:
    pm.DensityDist('pot1', logp=cust_logp, shape=(2,))

# %%
pm.set_tt_rng(42)
np.random.seed(42)
with pot1m:
    trace = pm.sample(1000, init='auto', cores=2, start=[dict(pot1=np.array([-2, 0])),
                                                         dict(pot1=np.array([2, 0]))])

# %%

into

# %%
fig, ax = plt.subplots(2, 2, figsize=(12, 12))
ax = ax.flatten()
contour_pot(pot1f, ax[0], 'pot1', );
contour_pot(pot2f, ax[1], 'pot2');
contour_pot(pot3f, ax[2], 'pot3');
contour_pot(pot4f, ax[3], 'pot4');
fig.tight_layout()


def cust_logp(z):
    #return bound(-pot1(z), z>-5, z<5)
    return -pot1(z)

with pm.Model() as pot1m:
    pm.DensityDist('pot1', logp=cust_logp, shape=(2,))

# %%
pm.set_tt_rng(42)
np.random.seed(42)
with pot1m:
    trace = pm.sample(1000, init='auto', cores=2, start=[dict(pot1=np.array([-2, 0])),
                                                         dict(pot1=np.array([2, 0]))])

# %%

[MarcoGorelli]

I don't think there's much we can do about a tool which removes # %% lines, as otherwise we have no way of separating cells. We could/should, though, raise a helpful error message if we fail to reconstruct the notebook

---

the error message should also redirect users to this GitHub page and encourage them to file a bug report

[MarcoGorelli]

reported here hakancelikdev/unimport#134

[MarcoGorelli]

closing for now as they need to fix their bug first