Function plot_wrapper

catboost/python-package/catboost/core.py:1568–1578

(plot, plot_file, plot_title, train_dirs)

@contextmanager
def plot_wrapper(plot, plot_file, plot_title, train_dirs):
    if plot:
        widget = _get_catboost_widget(train_dirs)
        widget._run_update()
    try:
        yield
    finally:
        if plot:
            widget._stop_update()
    if plot_file is not None:
        OfflineMetricVisualizer(train_dirs).save_to_file(plot_title, plot_file)


# the first element of the synonyms list is the canonical name