move test graph to ns pkg

wildlife-dynamics · Sep 10, 2024 · 6386072 · 6386072
1 parent d19ba12
commit 6386072
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py b/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py
@@ -3,8 +3,8 @@
 from graphlib import TopologicalSorter
 from typing import Any, Literal
 
-from ecoscope_workflows.decorators import AsyncTask
-from ecoscope_workflows.executors import Future, FutureSequence
+from ecoscope_workflows.core.decorators import AsyncTask
+from ecoscope_workflows.core.executors import Future, FutureSequence
 
 logger = logging.getLogger(__name__)
 

diff --git a/tests/test_graph.py → ...oscope-workflows/core/tests/test_graph.py b/tests/test_graph.py → ...oscope-workflows/core/tests/test_graph.py
@@ -1,14 +1,10 @@
 from dataclasses import dataclass
 from functools import lru_cache
 from typing import Sequence, TypeVar
-# from textwrap import dedent
 
-# import yaml
-
-# from ecoscope_workflows.compiler import Spec, TaskInstance
-from ecoscope_workflows.decorators import task
-from ecoscope_workflows.executors import Future, FutureSequence, LithopsExecutor
-from ecoscope_workflows.graph import DependsOn, DependsOnSequence, Graph, Node
+from ecoscope_workflows.core.decorators import task
+from ecoscope_workflows.core.executors import Future, FutureSequence, LithopsExecutor
+from ecoscope_workflows.core.graph import DependsOn, DependsOnSequence, Graph, Node
 
 T = TypeVar("T")