VIZ format table cells with markers - #993

from benchopt import BasePlot


class Plot(BasePlot):
    name = "Custom table"
    type = "table"
    options = {
        "dataset": ...,
    }

    def plot(self, df, dataset):
        df = df[df['dataset_name'] == dataset]
        # Last value reached by each solver, median over the repetitions
        results = (
            df.select_dtypes(include=['number'])
            .join(df['solver_name'])
            .groupby(['solver_name', 'stop_val'])
            .median()
            .groupby('solver_name')
            .last()
        )
        best = results['objective_value'].idxmin()

        return [
            [
                f"**{solver}**" if solver == best else solver,
                f"**{res['objective_value']}**" if solver == best else res['objective_value'],
                res['time'],
                "__best objective__" if solver == best else r"*runner-up\**",
            ]
            for solver, res in results.iterrows()
        ]

    def get_metadata(self, df, dataset):
        return {
            "title": f"Custom table\nData: {dataset}",
            "columns": ["solver", "objective_value", "time (s)", "comment"],
            "default_order_column": "objective_value",
            "default_order_ascending": True,
        }