fix: 🐛 preserve order of the columns when infered from the rows (#146)

huggingface · Feb 15, 2022 · a98f857 · a98f857
1 parent 93c218e
commit a98f857
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -2,7 +2,7 @@
 authors = ["Sylvain Lesage <[email protected]>"]
 description = "API to extract rows of 🤗 datasets"
 name = "datasets-preview-backend"
-version = "0.16.18"
+version = "0.16.19"
 
 [tool.poetry.dependencies]
 Pillow = "^8.4.0"

diff --git a/src/datasets_preview_backend/models/column/__init__.py b/src/datasets_preview_backend/models/column/__init__.py
@@ -75,7 +75,9 @@ def get_columns(info: Info, rows: List[Row]) -> List[Column]:
             return []
         else:
             column_names = list(
-                {column_name for row in rows[:MAX_ROWS_FOR_TYPE_INFERENCE_AND_CHECK] for column_name in row.keys()}
+                dict.fromkeys(
+                    column_name for row in rows[:MAX_ROWS_FOR_TYPE_INFERENCE_AND_CHECK] for column_name in row.keys()
+                )
             )
     else:
         column_names = list(features.keys())