snapshot

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)