Skip to content

Commit 6e9ead6

Browse files
committed
possible fix for failing groupby test
1 parent 21afbb1 commit 6e9ead6

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

xarray/core/concat.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -658,6 +658,11 @@ def get_indexes(name):
658658
for name, result_var in result_vars.items()
659659
if name in coord_names
660660
}
661+
result_data_vars = {
662+
name: result_var
663+
for name, result_var in result_vars.items()
664+
if name not in coord_names
665+
}
661666

662667
if index is not None:
663668
if dim_var is not None:
@@ -668,16 +673,14 @@ def get_indexes(name):
668673
coord_vars[dim] = index_vars[dim]
669674
result_indexes[dim] = index
670675
unlabeled_dims = unlabeled_dims - set([dim])
676+
else:
677+
if dim in result_data_vars:
678+
coord_vars[dim] = result_data_vars[dim]
679+
result_data_vars.pop(dim)
671680

672681
# TODO: add indexes at Dataset creation (when it is supported)
673682
coords = Coordinates(coord_vars, indexes=result_indexes)
674683

675-
result_data_vars = {
676-
name: result_var
677-
for name, result_var in result_vars.items()
678-
if name not in coord_names
679-
}
680-
681684
result = type(datasets[0])(result_data_vars, coords=coords, attrs=result_attrs)
682685
result.encoding = result_encoding
683686

0 commit comments

Comments
 (0)