f9d92551d9fa135ebc246d5b28b98331440093bc,mlflow/store/sqlalchemy_store.py,SqlAlchemyStore,_search_runs,#SqlAlchemyStore#Any#Any#Any#Any#Any#Any#,526

Before Change

                                  INVALID_PARAMETER_VALUE)
        with self.ManagedSessionMaker() as session:
            runs = [run.to_mlflow_entity(session)
                    for exp in experiment_ids
                    for run in self._list_runs(session, exp, run_view_type)]
            filtered = SearchUtils.filter(runs, filter_string)
            sorted_runs = SearchUtils.sort(filtered, order_by)

After Change

            // tags. These run attributes are referenced during the invocation of
            // ``run.to_mlflow_entity()``, so eager loading helps avoid additional database queries
            // that are otherwise executed at attribute access time under a lazy loading model.
            queried_runs = session \
                .query(SqlRun) \
                .options(*self._get_eager_run_query_options()) \
                .filter(
                    SqlRun.experiment_id.in_(experiment_ids),
                    SqlRun.lifecycle_stage.in_(stages)) \
                .all()
            runs = [run.to_mlflow_entity() for run in queried_runs]

        filtered = SearchUtils.filter(runs, filter_string)
        sorted_runs = SearchUtils.sort(filtered, order_by)

Instances