Decorators for Regularize step in mnist tutorial fail when run for the second time with addition of force_rerun = True 