File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -171,23 +171,6 @@ def _parse_schema(ds) -> pa.Schema:
171171 return pa .schema (columns )
172172
173173
174- def _parse_schema (ds ) -> pa .Schema :
175- """Extracts a `pa.Schema` from the Dataset, treating dims and data_vars as columns."""
176- columns = []
177-
178- for coord_name , coord_var in ds .coords .items ():
179- # Only include dimension coordinates
180- if coord_name in ds .dims :
181- pa_type = pa .from_numpy_dtype (coord_var .dtype )
182- columns .append (pa .field (coord_name , pa_type ))
183-
184- for var_name , var in ds .data_vars .items ():
185- pa_type = pa .from_numpy_dtype (var .dtype )
186- columns .append (pa .field (var_name , pa_type ))
187-
188- return pa .schema (columns )
189-
190-
191174def read_xarray (ds : xr .Dataset , chunks : Chunks = None ) -> pa .RecordBatchReader :
192175 """Pivots an Xarray Dataset into a PyArrow Table, partitioned by chunks.
193176
You can’t perform that action at this time.
0 commit comments