Fix empty table if not all results are available

Author: amaslenn

src/cloudai/report_generator/comparison_report.py

@@ -143,12 +143,13 @@ def create_table(
                     no_wrap=False,
                 )
 
-        for row_idx in range(len(dfs[0][info_columns[0]])):
+        df_with_max_rows = max(dfs, key=len)
+        for row_idx in range(len(df_with_max_rows[info_columns[0]])):
             data = []
             for df in dfs:
-                data.extend([str(df[col].get(row_idx)) for col in data_columns])
+                data.extend([str(df[col].get(row_idx, "n/a")) for col in data_columns])
 
-            table.add_row(*[str(dfs[0][col][row_idx]) for col in info_columns], *data)
+            table.add_row(*[str(df_with_max_rows[col][row_idx]) for col in info_columns], *data)
 
         return table
 

tests/report_generation_strategy/test_comparison_report.py

@@ -101,8 +101,8 @@ def test_one_data_point_is_empty(self, cmp_report: MyComparisonReport, nccl_tr:
                 ],
             ),
             [
-                pd.DataFrame({"size": [1, 2, 4], "value": [10, 20, 40]}),
                 pd.DataFrame({"size": [], "value": []}),
+                pd.DataFrame({"size": [1, 2, 4], "value": [10, 20, 40]}),
             ],
             "title",
             ["size"],
@@ -112,8 +112,8 @@ def test_one_data_point_is_empty(self, cmp_report: MyComparisonReport, nccl_tr:
         assert len(table.columns) == 3
         assert len(table.rows) == 3
         assert list(table.columns[0].cells) == ["1", "2", "4"]
-        assert list(table.columns[1].cells) == ["10", "20", "40"]
-        assert list(table.columns[2].cells) == ["None", "None", "None"]
+        assert list(table.columns[1].cells) == ["n/a", "n/a", "n/a"]
+        assert list(table.columns[2].cells) == ["10", "20", "40"]
 
 
 def test_create_charts(cmp_report: MyComparisonReport, nccl_tr: TestRun) -> None: