handle single numbers

pytorch · Jun 19, 2024 · 475a659 · 475a659
1 parent 466ea8e
commit 475a659
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ignite/utils.py b/ignite/utils.py
@@ -175,7 +175,6 @@ def flatten_to_str(data: Any) -> List[str]:
 
             ['tensor: Shape: torch.Size([2, 2])']
     """
-
     formatted_items: List[str] = []
 
     def format_item(item: Any, prefix: str = "") -> Optional[str]:
@@ -209,6 +208,10 @@ def format_item(item: Any, prefix: str = "") -> Optional[str]:
                 formatted_items.append(f"{prefix}")
         return None
 
+    # Directly handle single numeric values
+    if isinstance(data, numbers.Number):
+        return [f"{data:.4f}"]
+
     format_item(data)
     return formatted_items
 

diff --git a/tests/ignite/test_utils.py b/tests/ignite/test_utils.py
@@ -58,6 +58,7 @@ def test_convert_tensor():
 @pytest.mark.parametrize(
     "input_data,expected",
     [
+        (42, ["42.0000"]),
         ([{"a": 15, "b": torch.tensor([2.0])}], ["a: 15.0000", "b: [2.0000]"]),
         ({"a": 10, "b": 2.33333}, ["a: 10.0000", "b: 2.3333"]),
         ({"x": torch.tensor(0.1234), "y": [1, 2.3567]}, ["x: 0.1234", "y: 1.0000, 2.3567"]),