diff --git a/pel/core.py b/pel/core.py index 387e5f1..d2a7a5b 100644 --- a/pel/core.py +++ b/pel/core.py @@ -1,7 +1,7 @@ """The Pel Graph, Task, and ArgParser models.""" import argparse +import collections import threading -from collections import OrderedDict from enum import Enum, auto from typing import ( Any, @@ -10,6 +10,7 @@ List, Mapping, Optional, + OrderedDict, Sequence, Set, Type, @@ -139,7 +140,7 @@ def _get_needs_map( self, end_task_names: Union[Sequence[str], str] = "" ) -> OrderedDict[str, Set[str]]: """Return a dict mapping task names to their dependencies' names.""" - needs_map: OrderedDict[str, Set[str]] = OrderedDict() + needs_map: OrderedDict[str, Set[str]] = collections.OrderedDict() if end_task_names: tasks = self._get_recursive_need_names(end_task_names) else: