diff --git a/wsimod/orchestration/model.py b/wsimod/orchestration/model.py index 8e890175..50507249 100644 --- a/wsimod/orchestration/model.py +++ b/wsimod/orchestration/model.py @@ -157,6 +157,8 @@ def load(self, address, config_name="config.yml", overrides={}): config_name: overrides: """ + from ..extensions import apply_patches + with open(os.path.join(address, config_name), "r") as file: data = yaml.safe_load(file) @@ -191,6 +193,8 @@ def load(self, address, config_name="config.yml", overrides={}): if "dates" in data.keys(): self.dates = [to_datetime(x) for x in data["dates"]] + apply_patches(self) + def save(self, address, config_name="config.yml", compress=False): """Save the model object to a yaml file and input data to csv.gz format in the directory specified.