Move FreeFormFlows to experimental module

bayesflow/experimental/__init__.py

@@ -1,5 +1,6 @@
 from .cif import CIF
 from .continuous_time_consistency_model import ContinuousTimeConsistencyModel
+from .free_form_flow import FreeFormFlow
 
 from ..utils._docs import _add_imports_to_all
 

bayesflow/networks/free_form_flow/free_form_flow.py → bayesflow/experimental/free_form_flow/free_form_flow.py

@@ -14,7 +14,7 @@
     deserialize_value_or_type,
 )
 
-from ..inference_network import InferenceNetwork
+from bayesflow.networks import InferenceNetwork
 
 
 @serializable(package="networks.free_form_flow")

bayesflow/networks/__init__.py

@@ -2,7 +2,6 @@
 from .coupling_flow import CouplingFlow
 from .deep_set import DeepSet
 from .flow_matching import FlowMatching
-from .free_form_flow import FreeFormFlow
 from .inference_network import InferenceNetwork
 from .mlp import MLP
 from .lstnet import LSTNet

tests/test_networks/conftest.py

@@ -50,14 +50,14 @@ def coupling_flow_subnet(subnet):
 
 @pytest.fixture()
 def free_form_flow():
-    from bayesflow.networks import FreeFormFlow
+    from bayesflow.experimental import FreeFormFlow
 
     return FreeFormFlow()
 
 
 @pytest.fixture()
 def free_form_flow_subnet(subnet):
-    from bayesflow.networks import FreeFormFlow
+    from bayesflow.experimental import FreeFormFlow
 
     return FreeFormFlow(encoder_subnet=subnet, decoder_subnet=subnet)