fix bug for on_premises yaml

Author: peterdudfield

nowcasting_dataset/config/model.py

@@ -266,6 +266,14 @@ class Satellite(DataSourceMixin, TimeResolutionMixin):
         description="The number of meters per pixel for non-HRV satellite channels.",
     )
 
+    keep_dawn_dusk_hours: int = Field(
+        0,
+        description="The number hours around dawn and dusk that should be keep. "
+        "I.e 'keep_dawn_dusk_hours'=2,"
+        " then if dawn if 07.00, "
+        " then data is keep from 06.00",
+    )
+
 
 class HRVSatellite(DataSourceMixin, TimeResolutionMixin):
     """Satellite configuration model for HRV data"""

tests/config/test_config.py

@@ -35,6 +35,18 @@ def test_yaml_load():
     assert "gs://" in str(config.output_data.filepath)
 
 
+def test_yaml_load_on_premises():
+    """Test that yaml loading works for 'on_premises.yaml'"""
+
+    filename = os.path.join(
+        os.path.dirname(nowcasting_dataset.__file__), "config", "on_premises.yaml"
+    )
+
+    config = load_yaml_configuration(filename)
+
+    assert isinstance(config, Configuration)
+
+
 def test_yaml_save():
     """
     Check a configuration can be saved to a yaml file