From 6816de749ebfac66d841da302a05c8908d089c25 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Thu, 21 Mar 2024 10:42:28 +0100 Subject: [PATCH] fix to_csv --- optimum_benchmark/hub_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum_benchmark/hub_utils.py b/optimum_benchmark/hub_utils.py index 1218dd65..23b235c0 100644 --- a/optimum_benchmark/hub_utils.py +++ b/optimum_benchmark/hub_utils.py @@ -36,7 +36,7 @@ def to_json(self, path: str, flat: bool = False) -> None: def to_dataframe(self) -> pd.DataFrame: flat_report_dict = self.to_flat_dict() - return pd.DataFrame.from_dict(flat_report_dict, orient="index") + return pd.DataFrame.from_dict(flat_report_dict, orient="index").T def to_csv(self, path: str) -> None: self.to_dataframe().to_csv(path, index=False)