fix rows schema mismatch (#3125)

huggingface · Jan 6, 2025 · 5711204 · 5711204
1 parent 30ae84d
commit 5711204
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/libcommon/src/libcommon/parquet_utils.py b/libs/libcommon/src/libcommon/parquet_utils.py
@@ -147,7 +147,7 @@ class RowGroupReader:
     features: Features
 
     def read(self, columns: list[str]) -> pa.Table:
-        if not set(columns) <= set(self.parquet_file.schema_arrow.names):
+        if not set(self.parquet_file.schema_arrow.names) <= set(columns):
             raise SchemaMismatchError(
                 f"Parquet files have different columns: {sorted(columns)} and {sorted(self.parquet_file.schema_arrow.names)}"
             )