refactor: sys-model

SwanHubX · Dec 9, 2024 · 6cbb528 · 6cbb528
1 parent 0c1d1c0
commit 6cbb528
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "swankit"
-version = "0.1.2b2"
+version = "0.1.2b3"
 dynamic = ["readme", "dependencies"]
 description = "Base toolkit for SwanLab"
 license = "Apache-2.0"

diff --git a/swankit/callback/__init__.py b/swankit/callback/__init__.py
@@ -107,4 +107,4 @@ def __str__(self) -> str:
         pass
 
 
-__all__ = ["SwanKitCallback", "models"]
+__all__ = ["SwanKitCallback", "ColumnInfo", "MetricInfo", "ColumnConfig", "OperateErrorInfo", "RuntimeInfo"]
diff --git a/swankit/callback/models/key.py b/swankit/callback/models/key.py
@@ -18,12 +18,21 @@
 YRange = Optional[Tuple[Optional[float], Optional[float]]]
 
 
-class ColumnConfig(TypedDict):
+class ColumnConfig:
     """
     列信息配置
     """
 
-    y_range: YRange
+    def __init__(self, y_range: YRange = None, chart_name: Optional[str] = None, metric_name: Optional[str] = None):
+        """
+        生成的列信息配置对象
+        :param y_range: y轴范围
+        :param chart_name: 图表名称
+        :param metric_name: 指标名称
+        """
+        self.y_range: YRange = y_range
+        self.chart_name: Optional[str] = chart_name
+        self.metric_name: Optional[str] = metric_name
 
 
 class ColumnInfo:
Original file line number	Diff line number	Diff line change
Expand Up		@@ -107,4 +107,4 @@ def __str__(self) -> str:
		pass


		__all__ = ["SwanKitCallback", "models"]
		__all__ = ["SwanKitCallback", "ColumnInfo", "MetricInfo", "ColumnConfig", "OperateErrorInfo", "RuntimeInfo"]