reformatting

Signed-off-by: JoseSantosAMD <Jose.Santos@amd.com>
ROCm · Aug 2, 2023 · 80e6583 · 80e6583
1 parent efb1398
commit 80e6583
Showing 1 changed file with 107 additions and 14 deletions.
diff --git a/src/omniperf_analyze/utils/tty.py b/src/omniperf_analyze/utils/tty.py
@@ -150,28 +150,121 @@ def show_all(args, runs, archConfigs, output):
                                             for x in base_df[header]
                                         ]
                                         if "Unit" in cur_df.columns:
-                                            for idx, row in cur_df[cur_df["Unit"] == "Gb/s"].items():
+                                            for idx, row in cur_df[
+                                                cur_df["Unit"] == "Gb/s"
+                                            ].items():
                                                 for curr_metric in row:
-                                                    curr_row = cur_df[cur_df["Metric"] == curr_metric]
+                                                    curr_row = cur_df[
+                                                        cur_df["Metric"] == curr_metric
+                                                    ]
                                                     if not curr_row.empty:
                                                         if "Value" in curr_row:
                                                             if curr_row.Value[0] < 0.001:
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Unit"] = "Kb/s"
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Value"] = 1000000* curr_row.Value
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Unit",
+                                                                ] = "Kb/s"
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Value",
+                                                                ] = (
+                                                                    1000000
+                                                                    * curr_row.Value
+                                                                )
                                                             elif curr_row.Value[0] < 1:
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Unit"] = "Mb/s"
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Value"] = 1000 * curr_row.Value
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Unit",
+                                                                ] = "Mb/s"
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Value",
+                                                                ] = (
+                                                                    1000 * curr_row.Value
+                                                                )
                                                         elif "Avg" in curr_row:
                                                             if curr_row.Avg[0] < 0.001:
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Unit"] = "Kb/s"
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Avg"] = 1000000* curr_row.Avg
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Min"] = 1000000* curr_row.Min
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Max"] = 1000000* curr_row.Max
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Unit",
+                                                                ] = "Kb/s"
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Avg",
+                                                                ] = (
+                                                                    1000000 * curr_row.Avg
+                                                                )
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Min",
+                                                                ] = (
+                                                                    1000000 * curr_row.Min
+                                                                )
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Max",
+                                                                ] = (
+                                                                    1000000 * curr_row.Max
+                                                                )
                                                             elif curr_row.Avg[0] < 1:
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Unit"] = "Mb/s"
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Avg"] = 1000 * curr_row.Avg
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Min"] = 1000 * curr_row.Min
-                                                                cur_df.loc[(cur_df["Metric"] == curr_metric), "Max"] = 1000 * curr_row.Max
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Unit",
+                                                                ] = "Mb/s"
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Avg",
+                                                                ] = (
+                                                                    1000 * curr_row.Avg
+                                                                )
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Min",
+                                                                ] = (
+                                                                    1000 * curr_row.Min
+                                                                )
+                                                                cur_df.loc[
+                                                                    (
+                                                                        cur_df["Metric"]
+                                                                        == curr_metric
+                                                                    ),
+                                                                    "Max",
+                                                                ] = (
+                                                                    1000 * curr_row.Max
+                                                                )
                                         df = pd.concat([df, cur_df[header]], axis=1)
 
                 if not df.empty: