fix: import get_task_directories from utils

graeter-group · Mar 25, 2024 · 8d4ba4d · 8d4ba4d
1 parent 40ee0d2
commit 8d4ba4d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/kimmdy/runmanager.py b/src/kimmdy/runmanager.py
@@ -21,7 +21,6 @@
 from typing import Optional
 import re
 
-from kimmdy.analysis import get_task_directories
 from kimmdy.config import Config
 from kimmdy.constants import MARK_STARTED, MARK_DONE, MARK_FAILED, MARKERS
 from kimmdy.coordinates import break_bond_plumed, merge_top_slow_growth, place_atom
@@ -37,7 +36,7 @@
 from kimmdy.tasks import Task, TaskFiles, get_plumed_out
 from kimmdy.topology.topology import Topology
 from kimmdy.topology.utils import get_is_reactive_predicate_f
-from kimmdy.utils import run_gmx, truncate_sim_files
+from kimmdy.utils import run_gmx, truncate_sim_files, get_task_directories
 
 logger = logging.getLogger(__name__)
 

diff --git a/tests/test_config.py b/tests/test_config.py
@@ -171,6 +171,7 @@ def test_get_existing_files(arranged_tmp_path):
             "broken_equil_f1000.mdp",
             "edissoc.dat",
             "ffbonded.itp",
+            "yml",
         ]
     )
 

diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -9,7 +9,7 @@
 from kimmdy.parsing import read_top, write_top
 from kimmdy.topology.topology import Topology
 from kimmdy.plugins import parameterization_plugins
-from kimmdy.analysis import get_task_directories
+from kimmdy.utils import get_task_directories
 
 
 def read_last_line(file):