feat: format metric obj

swankit/callback/models/key.py

@@ -211,14 +211,20 @@ def column_error(self) -> Optional[ParseErrorInfo]:
         return self.column_info.error
 
     @property
-    def data(self) -> Union[Dict, None]:
+    def data(self):
         """
         指标数据的data字段
         """
         if self.is_error:
             return None
         return self.metric["data"]
 
+    def __str__(self):
+        return str(self.data)
+
+    def __repr__(self):
+        return str(self.data)
+
 
 class MetricErrorInfo(MetricInfo):
     def __init__(self, column_info: ColumnInfo, error: ParseErrorInfo):

test/unit/callback/models/test_key.py

@@ -109,3 +109,5 @@ def test_metric_info():
     assert m.metric_epoch == 1
     assert m.swanlab_media_dir == "."
     assert m.metric_file_path == f"./{c.kid}/1.log"
+    assert str(m) == "1"
+    assert repr(m) == "1"