diff --git a/modules/util/split_nc4.py b/modules/util/split_nc4.py index 9480d0384cd64d336b02ec569400cbdc45fa3f40..9b1bf38d8100ee15d22faad63249848564cba9f0 100644 --- a/modules/util/split_nc4.py +++ b/modules/util/split_nc4.py @@ -5,16 +5,16 @@ import numpy as np def split_dataset(input_file, output_pattern, dim_name, chunk_size): # Load the input dataset ds = nc.Dataset(input_file, 'r', format='NETCDF4') - dim_size = len(ds.dimensions[dim_name]) + outer_dim_size = len(ds.dimensions[dim_name]) # Calculate the number of chunks - num_chunks = int(np.ceil(dim_size / chunk_size)) + num_chunks = int(np.ceil(outer_dim_size / chunk_size)) # Loop through each chunk for i in range(num_chunks-1): # Determine the start and end indices of this chunk start = i * chunk_size - end = min((i + 1) * chunk_size, dim_size) + end = min((i + 1) * chunk_size, outer_dim_size) # Slicing along our dimension of interest slice_indices = slice(start, end)