9163c404a731ce79899287c6e4709c6f9c900f55,mlflow/store/sqlalchemy_store.py,SqlAlchemyStore,list_run_infos,#SqlAlchemyStore#Any#Any#,232

Before Change

        raise NotImplementedError()

    def list_run_infos(self, experiment_id, _=None):
        exp = self.session.query(SqlExperiment).get(experiment_id)
        infos = []
        for run in exp.runs:
            infos.append(run.to_mlflow_entity().info)
        return infos

After Change

        return [run for run in exp.runs if run.lifecycle_stage in stages]

    def list_run_infos(self, experiment_id, run_view_type):
        return [r.to_mlflow_entity().info for r in self._list_runs(experiment_id, run_view_type)]

Instances