Przeglądaj źródła

修复图片数据集下载无记录问题

lxylxy123321 11 godzin temu
rodzic
commit
d6d978de5b
1 zmienionych plików z 6 dodań i 0 usunięć
  1. 6 0
      backend/app/services/dataset_service.py

+ 6 - 0
backend/app/services/dataset_service.py

@@ -324,6 +324,12 @@ def _download_modelscope_dataset(dataset_id: str) -> tuple[Path, Path, int]:
                     records.append(json.loads(line))
                 except json.JSONDecodeError:
                     continue
+    elif target.suffix == ".csv":
+        import csv as _csv
+        records = []
+        reader = _csv.DictReader(content.splitlines())
+        for row in reader:
+            records.append(dict(row))
     else:
         records = []