Skip to content

Commit

Permalink
Fix for higher dimensional sample point arrays
Browse files Browse the repository at this point in the history
  • Loading branch information
duncanwp committed Aug 8, 2016
1 parent 57e6337 commit 539597f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions cis/collocation/gridded_interpolation.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,11 +80,11 @@ def __init__(self, _data, sample, method='linear', missing_data_for_missing_samp
if len(data.coords('altitude', dim_coords=False)) > 0 and sample.coords('altitude'):
hybrid_coord = data.coord('altitude').points
hybrid_dims = data.coord_dims(data.coord('altitude'))
sample_points.append(sample.coord("altitude").points)
sample_points.append(sample.coord("altitude").data_flattened)
elif len(data.coords('air_pressure', dim_coords=False)) > 0 and sample.coords('air_pressure'):
hybrid_coord = data.coord('air_pressure').points
hybrid_dims = data.coord_dims(data.coord('air_pressure'))
sample_points.append(sample.coord("air_pressure").points)
sample_points.append(sample.coord("air_pressure").data_flattened)
else:
hybrid_coord = None
hybrid_dims = None
Expand All @@ -95,7 +95,7 @@ def __init__(self, _data, sample, method='linear', missing_data_for_missing_samp

# Check if we want to sample missing points
if missing_data_for_missing_sample and hasattr(sample.data, 'mask'):
self.missing_mask = sample.data.mask
self.missing_mask = sample.data.mask.flatten()
sample_points = [p[~self.missing_mask] for p in sample_points]
else:
self.missing_mask = None
Expand Down

0 comments on commit 539597f

Please sign in to comment.