snapshot...

modules/contrail/util.py

@@ -95,10 +95,10 @@ def extract(mask_image, image_ts, clavrx_path):
     lon_range = [np.min(contrail_lons), np.max(contrail_lons)]
     lat_range = [np.min(contrail_lats), np.max(contrail_lats)]
 
-    uwind_3d = get_volume(xr_dataset, 'u-wind', 'm s-1', lon_range=[lon_range[0], lon_range[1]], lat_range=[lat_range[0], lat_range[1]])
-    vwind_3d = get_volume(xr_dataset, 'v-wind', 'm s-1', lon_range=[lon_range[0], lon_range[1]], lat_range=[lat_range[0], lat_range[1]])
-    temp_3d = get_volume(xr_dataset, 'temperature', 'degK', lon_range=[lon_range[0], lon_range[1]], lat_range=[lat_range[0], lat_range[1]])
-    rh_3d = get_volume(xr_dataset, 'rh', '%', lon_range=[lon_range[0], lon_range[1]], lat_range=[lat_range[0], lat_range[1]])
+    uwind_3d = get_volume(xr_dataset, 'u-wind', 'm s-1', lon_range=lon_range, lat_range=lat_range)
+    vwind_3d = get_volume(xr_dataset, 'v-wind', 'm s-1', lon_range=lon_range, lat_range=lat_range)
+    temp_3d = get_volume(xr_dataset, 'temperature', 'degK', lon_range=lon_range, lat_range=lat_range)
+    rh_3d = get_volume(xr_dataset, 'rh', '%', lon_range=lon_range, lat_range=lat_range)
 
     uwind_3d = uwind_3d.transpose('Pressure', 'Latitude', 'Longitude')
     vwind_3d = vwind_3d.transpose('Pressure', 'Latitude', 'Longitude')
@@ -112,7 +112,7 @@ def extract(mask_image, image_ts, clavrx_path):
     # This one's a bit more work: `first_derivative` only returns a ndarray with no units, so we use the
     # helper function to create a DataArray and add units via metpy's pint support
     vert_shear_3d = first_derivative(horz_wind_spd_3d, axis=0, x=temp_3d.coords['Pressure'])
-    vert_shear_3d = volume_np_to_xr(vert_shear_3d, ['Pressure', 'Latitude', 'Longitude'], lon_range=[lon_range[0], lon_range[1]], lat_range=[lat_range[0], lat_range[1]])
+    vert_shear_3d = volume_np_to_xr(vert_shear_3d, ['Pressure', 'Latitude', 'Longitude'], lon_range=lon_range, lat_range=lat_range)
     vert_shear_3d = vert_shear_3d / units.hPa
 
     voxel_dict = {key: [] for key in bins_dict.keys()}