Merge pull request #955 from OpenFreeEnergy/fix_slow_cli_call

move slow import inside function call
OpenFreeEnergy · Oct 12, 2024 · 83028b1 · 83028b1
2 parents 29d8cd9 + 0186b3c
commit 83028b1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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("")