add flow matching to tests

bayesflow/experimental/networks/flow_matching/flow_matching.py

@@ -17,6 +17,13 @@ def __init__(self, network: keras.Layer, **kwargs):
         super().__init__(**kwargs)
         self.network = network
 
+    @classmethod
+    def new(cls, network: str = "resnet", base_distribution: str = "normal"):
+        # TODO: we probably want to provide a factory method like this, since the other networks use it
+        #  for high-level input parameters
+        # network = find_network(network)
+        return cls(network, base_distribution=base_distribution)
+
     @classmethod
     def from_config(cls, config: dict, custom_objects=None) -> "FlowMatching":
         # TODO: the base distribution must be savable and loadable

tests/test_networks/conftest.py

@@ -14,6 +14,12 @@ def coupling_flow():
     return CouplingFlow.new()
 
 
+@pytest.fixture()
+def flow_matching():
+    from bayesflow.experimental.networks import FlowMatching
+    return FlowMatching.new()
+
+
 @pytest.fixture(params=["coupling_flow"])
 def inference_network(request):
     return request.getfixturevalue(request.param)