from glob import glob
import sys
sys.path.append("misc")
from wrap_try_accept import apply_fun_to_notebook, remove_superimport, wrap_try_accept_in_code, replace_pyprobml_utils
book2_nb = glob("notebooks/book2/*/*.ipynb")
for nb in book2_nb:
apply_fun_to_notebook(nb, replace_pyprobml_utils)