From c94d8a8397bf8d609f5b886cecde2ad16eb0280b Mon Sep 17 00:00:00 2001 From: Magnus Ulimoen Date: Thu, 23 Jan 2025 14:43:06 +0100 Subject: [PATCH] Rename to include Concat --- src/pyaro_readers/merging_reader.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/pyaro_readers/merging_reader.py b/src/pyaro_readers/merging_reader.py index d95c0cd..0da498b 100644 --- a/src/pyaro_readers/merging_reader.py +++ b/src/pyaro_readers/merging_reader.py @@ -18,7 +18,7 @@ class MergingReaderException(Exception): pass -class MergingReaderData(Data): +class MergingReaderConcatData(Data): def __init__(self, data: list[Data], variable: str) -> None: if len(data) == 0: raise MergingReaderException("Requires at least one dataset") @@ -40,7 +40,7 @@ def keys(self): raise NotImplementedError def slice(self, index): - return MergingReaderData([d[index] for d in self._data], self._variable) + return MergingReaderConcatData([d[index] for d in self._data], self._variable) @property def values(self) -> np.ndarray: @@ -90,6 +90,7 @@ def __init__( raise MergingReaderException( 'Only merging mode "concat" is supported as of now' ) + self._mode = mode self._datasets = [] self._set_filters(filters) for d in datasets: @@ -99,7 +100,9 @@ def __init__( ) def _unfiltered_data(self, varname: str) -> Data: - return MergingReaderData([d.data(varname) for d in self._datasets], varname) + return MergingReaderConcatData( + [d.data(varname) for d in self._datasets], varname + ) def _unfiltered_stations(self) -> list[str]: stations = []