Merge pull request #654 from openclimatefix/fake-optical-flow

option to not make optical flow when making fake batch

Author: peterdudfield

nowcasting_dataset/data_sources/fake/batch.py

@@ -396,6 +396,9 @@ def optical_flow_fake(
         configuration = Configuration()
         configuration.input_data = Configuration().input_data.set_all_to_defaults()
 
+    if configuration.input_data.opticalflow is None:
+        return None
+
     batch_size = configuration.process.batch_size
     image_size_pixels_height = (
         configuration.input_data.opticalflow.opticalflow_input_image_size_pixels_height

tests/data_sources/fake/test_fake.py

@@ -37,6 +37,13 @@ def test_metadata_fake_gsp():
     assert m.y_centers_osgb[0] in metadata["location_y"].values
 
 
+def test_model_no_opticalflow(configuration):  # noqa: D103
+
+    configuration.input_data.opticalflow = None
+
+    _ = Batch.fake(configuration=configuration)
+
+
 def test_model(configuration):  # noqa: D103
 
     assert configuration.input_data.opticalflow is not None