diff --git a/openfecli/commands/plan_rbfe_network.py b/openfecli/commands/plan_rbfe_network.py index 2c2d13fdd..88beb4007 100644 --- a/openfecli/commands/plan_rbfe_network.py +++ b/openfecli/commands/plan_rbfe_network.py @@ -7,8 +7,6 @@ from openfecli.parameters import ( MOL_DIR, PROTEIN, OUTPUT_DIR, COFACTORS, YAML_OPTIONS, ) -from openfecli.plan_alchemical_networks_utils import plan_alchemical_network_output - def plan_rbfe_network_main( mapper, @@ -116,6 +114,8 @@ def plan_rbfe_network( which is detailed in the Options section. For more advanced setups, please consider using the Python layer of openfe. """ + from openfecli.plan_alchemical_networks_utils import plan_alchemical_network_output + write("RBFE-NETWORK PLANNER") write("______________________") write("") diff --git a/openfecli/commands/plan_rhfe_network.py b/openfecli/commands/plan_rhfe_network.py index fe5bd3114..90699d322 100644 --- a/openfecli/commands/plan_rhfe_network.py +++ b/openfecli/commands/plan_rhfe_network.py @@ -10,8 +10,6 @@ from openfecli.parameters import ( MOL_DIR, MAPPER, OUTPUT_DIR, YAML_OPTIONS, ) -from openfecli.plan_alchemical_networks_utils import plan_alchemical_network_output - def plan_rhfe_network_main( mapper, mapping_scorer, ligand_network_planner, small_molecules, @@ -100,6 +98,9 @@ def plan_rhfe_network(molecules: List[str], yaml_settings: str, output_dir: str) transformation a JSON file, that can be run with quickrun (or other future tools). """ + + from openfecli.plan_alchemical_networks_utils import plan_alchemical_network_output + write("RHFE-NETWORK PLANNER") write("______________________") write("")