diff --git a/mlos_bench/mlos_bench/environments/local/local_env.py b/mlos_bench/mlos_bench/environments/local/local_env.py index 38a09c72d1..2f72fc1f16 100644 --- a/mlos_bench/mlos_bench/environments/local/local_env.py +++ b/mlos_bench/mlos_bench/environments/local/local_env.py @@ -155,7 +155,7 @@ def run(self) -> Tuple[Status, Optional[dict]]: self._read_results_file, extra_paths=[temp_dir])) _LOG.debug("Read data:\n%s", data) - if len(data) != 1: + if list(data.columns) == ["metric", "value"]: _LOG.warning( "Local run has %d rows: assume long format of (metric, value)", len(data)) data = pandas.DataFrame([data.value.to_list()], columns=data.metric.to_list())