File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1919
2020default_logger = logging .getLogger (__name__ )
2121
22+
2223def stitchee (
2324 files_to_concat : list [str ],
2425 output_file : str ,
@@ -125,10 +126,12 @@ def stitchee(
125126
126127 tree_dicts = [tree .to_dict () for tree in xrdatatree_list ]
127128 keys_list = [set (t .keys ()) for t in tree_dicts ]
128- symmetric_diff = any ( [kl ^ keys_list [0 ] for kl in keys_list ] )
129+ symmetric_diff = any ([kl ^ keys_list [0 ] for kl in keys_list ])
129130
130131 if symmetric_diff :
131- raise KeyError (f"Datatrees do not have matching Dataset nodes. Nodes that do not match: { symmetric_diff } ." )
132+ raise KeyError (
133+ f"Datatrees do not have matching Dataset nodes. Nodes that do not match: { symmetric_diff } ."
134+ )
132135
133136 # Files are concatenated together (Using XARRAY).
134137 start_time = time .time ()
@@ -147,7 +150,7 @@ def stitchee(
147150
148151 if concat_method == "xarray-concat" :
149152 combined_dict = {
150- kk : xr .concat (
153+ kk : xr .concat (
151154 [tree_dict [kk ] for tree_dict in tree_dicts ],
152155 data_vars = "minimal" ,
153156 coords = "minimal" ,
@@ -159,7 +162,7 @@ def stitchee(
159162
160163 elif concat_method == "xarray-combine" :
161164 combined_dict = {
162- kk : xr .combine_by_coords (
165+ kk : xr .combine_by_coords (
163166 [tree_dict [kk ] for tree_dict in tree_dicts ],
164167 data_vars = "minimal" ,
165168 coords = "minimal" ,
You can’t perform that action at this time.
0 commit comments