In note book the mlflow tagas ias the command travels and once failed or exit there itself it logs and finishes the experiment even if the noteboolsfails. However, if you want to continue logging metrics or artifacts to that run, you just need to use
mlflow.start_run(run_id="YourRunIDYouCanGetItFromUI")
.