Skip to content

Commit

Permalink
fixed nc topo exo handler which needs to have same init args and kwar…
Browse files Browse the repository at this point in the history
…gs for signature parsing
  • Loading branch information
grantbuster committed Feb 2, 2024
1 parent 6d0f100 commit cd82016
Showing 1 changed file with 17 additions and 17 deletions.
34 changes: 17 additions & 17 deletions sup3r/preprocessing/data_handling/exo_extraction.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,6 +148,9 @@ def __init__(self,
self.shape = shape
self.res_kwargs = res_kwargs

# for subclasses
self._source_handler = None

if input_handler is None:
in_type = get_source_type(file_paths)
if in_type == 'nc':
Expand Down Expand Up @@ -566,23 +569,20 @@ def get_cache_file(self, feature, s_enhance, t_enhance, t_agg_factor):
class TopoExtractNC(TopoExtractH5):
"""TopoExtract for netCDF files"""

def __init__(self, *args, **kwargs):
"""Parameters
----------
args : list
Same positional arguments as TopoExtract
kwargs : dict
Same keyword arguments as TopoExtract
"""
super().__init__(*args, **kwargs)
logger.info('Getting topography for full domain from '
f'{self._exo_source}')
self.source_handler = DataHandlerNC(
self._exo_source,
features=['topography'],
worker_kwargs={'ti_workers': self.ti_workers},
val_split=0.0,
)
@property
def source_handler(self):
"""Get the DataHandlerNC object that handles the .nc source topography
data file."""
if self._source_handler is None:
logger.info('Getting topography for full domain from '
f'{self._exo_source}')
self._source_handler = DataHandlerNC(
self._exo_source,
features=['topography'],
worker_kwargs={'ti_workers': self.ti_workers},
val_split=0.0,
)
return self._source_handler

@property
def source_data(self):
Expand Down

0 comments on commit cd82016

Please sign in to comment.