From 4f7178e9560207bfecc59bcd1bd72c895cc008d4 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 10 Jul 2024 10:35:50 -0500 Subject: [PATCH] snapshot --- modules/util/split_nc4.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/util/split_nc4.py b/modules/util/split_nc4.py index 9480d038..9b1bf38d 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) -- GitLab