From 5cafe746c4d6cb3ab0b9a520ee8ad7bfe170d1ac Mon Sep 17 00:00:00 2001
From: Nexisato <978452096@qq.com>
Date: Thu, 11 Jun 2026 17:00:38 +0800
Subject: [PATCH] fix: video column type compatible

---
 swanlab/sdk/internal/run/transforms/video/__init__.py | 3 ++-
 tests/unit/sdk/internal/run/transforms/test_video.py  | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/swanlab/sdk/internal/run/transforms/video/__init__.py b/swanlab/sdk/internal/run/transforms/video/__init__.py
index 54708955c..63d2ff201 100644
--- a/swanlab/sdk/internal/run/transforms/video/__init__.py
+++ b/swanlab/sdk/internal/run/transforms/video/__init__.py
@@ -99,7 +99,8 @@ def __init__(self, data_or_path: VideoDataType, caption: CaptionType = None):
 
     @classmethod
     def column_type(cls) -> ColumnType:
-        return ColumnType.COLUMN_TYPE_VIDEO
+        # TODO: 服务端支持 VIDEO 列类型后，改回 ColumnType.COLUMN_TYPE_VIDEO
+        return ColumnType.COLUMN_TYPE_IMAGE
 
     def transform(self, *, step: int, path: Path) -> MediaItem:
         content = self.buffer.getvalue()
diff --git a/tests/unit/sdk/internal/run/transforms/test_video.py b/tests/unit/sdk/internal/run/transforms/test_video.py
index 11804308f..0a2031622 100644
--- a/tests/unit/sdk/internal/run/transforms/test_video.py
+++ b/tests/unit/sdk/internal/run/transforms/test_video.py
@@ -118,7 +118,8 @@ class TestVideoColumnType:
     def test_column_type(self):
         from swanlab.proto.swanlab.metric.column.v1.column_pb2 import ColumnType
 
-        assert Video.column_type() == ColumnType.COLUMN_TYPE_VIDEO
+        # TODO: 服务端 column 暂时不支持 VIDEO，临时回退为 IMAGE
+        assert Video.column_type() == ColumnType.COLUMN_TYPE_IMAGE
 
 
 class TestVideoBuildDataRecord: