example of next(, None) (#24345)

2025-08-22 19:32:22 +09:00
parent 2a43e634e8
commit 51cc2bf429
1 changed files with 2 additions and 3 deletions
--- a/api/core/rag/extractor/excel_extractor.py
+++ b/api/core/rag/extractor/excel_extractor.py
@@ -34,9 +34,8 @@ class ExcelExtractor(BaseExtractor):
            for sheet_name in wb.sheetnames:
                sheet = wb[sheet_name]
                data = sheet.values
-                try:
-                    cols = next(data)
-                except StopIteration:
+                cols = next(data, None)
+                if cols is None:
                    continue
                df = pd.DataFrame(data, columns=cols)