]
PARAMETERS = []
PARAMETERS += [(path, mb) for path in glob("./notebooks/*.ipynb")
for mb in [None, 4]]
PARAMETERS += [(path, None) for path in glob("./../../examples/tutorials/*.ipynb")
if path.split("/")[-1][:2] in ALLOWED_TUTORIALS]
After Change
NOTEBOOKS_DIR = "./notebooks/"
NOTEBOOKS = glob(NOTEBOOKS_DIR + "*.ipynb")
TUTORIALS_DIR = "./../../examples/tutorials/"
TUTORIALS = glob(TUTORIALS_DIR + "*.ipynb")
ALLOWED_TUTORIALS = [
"01",
"02", // quite long
// "03", // very long
"04",
"07",
// "10", // requires multiprocess module
]
MICROBATCH_LIST = [None, 4]
DEVICE_LIST = [None, pytest.param("GPU:*", marks=NO_GPU)]
// Each parameter is (path, microbatch) configuration
PARAMETERS = []
// Run every notebook in test directory for every combination of microbatching
PARAMETERS += [(path, mb) for path in NOTEBOOKS
for mb in MICROBATCH_LIST]
// Run selected notebooks inside tutorials dir without microbatching
PARAMETERS += [(path, None) for path in TUTORIALS
if path.split("/")[-1][:2] in ALLOWED_TUTORIALS]
_ = [print(item) for item in PARAMETERS]
BAD_PREFIXES = ["get_ipython", "plt", "plot", "figure", "ax.",]
