diff --git a/tests/helper.py b/tests/helper.py index 0d2c834..e37e283 100644 --- a/tests/helper.py +++ b/tests/helper.py @@ -8,7 +8,7 @@ from __future__ import annotations -from functools import cache +from functools import lru_cache import numpy as np import xarray as xr @@ -47,7 +47,7 @@ def is_3d_rmse_better(result, obsp, simp) -> bool: return (rmse_values_new_ds < rmse_values_old_ds).all() -@cache +@lru_cache(maxsize=None) def get_datasets(kind: str) -> tuple[xr.Dataset, xr.Dataset, xr.Dataset, xr.Dataset]: historical_time = xr.cftime_range( "1971-01-01",