fix test

linshokaku · Jun 3, 2024 · a0cbe73 · a0cbe73
1 parent f30f0f4
commit a0cbe73
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_pfn_extras/training/extensions/print_report_notebook.py b/pytorch_pfn_extras/training/extensions/print_report_notebook.py
@@ -56,4 +56,7 @@ def __call__(self, manager: ExtensionsManagerProtocol) -> None:
         if self._infer_entries:
             # --- update entries ---
             self._update_entries(log_report)
+        for entry in self._entries:
+            if entry not in df.columns:
+                df[entry] = None
         self._widget.value = df[self._entries].to_html(index=False, na_rep="")