export GraphicalSimulator

Author: daniel-habermann

bayesflow/experimental/__init__.py

@@ -2,11 +2,11 @@
 Unstable or largely untested networks, proceed with caution.
 """
 
+from ..utils._docs import _add_imports_to_all
 from .cif import CIF
 from .continuous_time_consistency_model import ContinuousTimeConsistencyModel
 from .diffusion_model import DiffusionModel
 from .free_form_flow import FreeFormFlow
-
-from ..utils._docs import _add_imports_to_all
+from .graphical_simulator import GraphicalSimulator
 
 _add_imports_to_all(include_modules=["diffusion_model"])

bayesflow/experimental/graphical_simulator/__init__.py

@@ -0,0 +1 @@
+from .graphical_simulator import GraphicalSimulator