diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py index f749f53641755f204a739905a36418466e3f37d4..054713481478826af2c5220f2b9d9e9c08c4a0c2 100644 --- a/mlair/data_handler/data_handler_single_station.py +++ b/mlair/data_handler/data_handler_single_station.py @@ -63,7 +63,7 @@ class DataHandlerSingleStation(AbstractDataHandler): interpolation_method: Union[str, Tuple[str]] = DEFAULT_INTERPOLATION_METHOD, overwrite_local_data: bool = False, transformation=None, store_data_locally: bool = True, min_length: int = 0, start=None, end=None, variables=None, data_origin: Dict = None, - lazy_preprocessing: bool = False, **kwargs): + lazy_preprocessing: bool = False, overwrite_lazy_data=False, **kwargs): super().__init__() self.station = helpers.to_list(station) self.path = self.setup_data_path(data_path, sampling) @@ -94,6 +94,7 @@ class DataHandlerSingleStation(AbstractDataHandler): self.interpolation_method = interpolation_method self.overwrite_local_data = overwrite_local_data + self.overwrite_lazy_data = True if self.overwrite_local_data is True else overwrite_lazy_data self.store_data_locally = store_data_locally self.min_length = min_length self.start = start @@ -271,7 +272,7 @@ class DataHandlerSingleStation(AbstractDataHandler): hash = self._get_hash() filename = os.path.join(self.lazy_path, hash + ".pickle") try: - if self.overwrite_local_data is True: + if self.overwrite_lazy_data is True: raise FileNotFoundError with open(filename, "rb") as pickle_file: lazy_data = dill.load(pickle_file)