neural_nets to neural_networks

Author: BSnelling

petab/v2/core.py

@@ -1445,7 +1445,7 @@ def from_yaml(
                     nn_model_id=nn_id,
                 )
                 for nn_id, nn_config in (
-                    config.extensions[C.SCIML].neural_nets or {}
+                    config.extensions[C.SCIML].neural_networks or {}
                 ).items()
             ]
             if config.extensions and config.extensions[C.SCIML]
@@ -2719,7 +2719,7 @@ class SciMLConfig(BaseModel):
     # Absolute or relative to `base_path`.
     hybridization_files: list[AnyUrl | Path] = []
     #: The neural network IDs and info.
-    neural_nets: dict[str, NeuralNetConfig] | None = {}
+    neural_networks: dict[str, NeuralNetConfig] | None = {}
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -2840,9 +2840,9 @@ def to_yaml(self, filename: str | Path):
                 # convert Paths to strings
                 for key in ("array_files", "hybridization_files"):
                     d_ext[key] = list(map(str, d_ext[key]))
-                for nn in d_ext["neural_nets"]:
-                    d_ext["neural_nets"][nn][C.MODEL_LOCATION] = str(
-                        d_ext["neural_nets"][nn][C.MODEL_LOCATION]
+                for nn in d_ext["neural_networks"]:
+                    d_ext["neural_networks"][nn][C.MODEL_LOCATION] = str(
+                        d_ext["neural_networks"][nn][C.MODEL_LOCATION]
                     )
 
         write_yaml(data, filename)

tests/v2/test_sciml.py

@@ -25,7 +25,7 @@ def _get_test_problem():
                 "version": "0.1.0",
                 "array_files": ["net1_ps.hdf5"],
                 "hybridization_files": ["hybridizations.tsv"],
-                "neural_nets": {
+                "neural_networks": {
                     "net1": NeuralNetConfig(
                         location="net1.yaml",
                         pre_initialization=False,