Fix import

kedro-org · Apr 18, 2020 · 2b6efd7 · 2b6efd7
1 parent 981a223
commit 2b6efd7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/package/tests/test_server.py b/package/tests/test_server.py
@@ -41,9 +41,8 @@
 from toposort import CircularDependencyError
 
 from kedro_viz import server
-from kedro_viz.server import _allocate_port, format_pipeline_data
+from kedro_viz.server import _allocate_port, _sort_layers, format_pipeline_data
 from kedro_viz.utils import WaitForException
-from package.kedro_viz.server import _sort_layers
 
 EXPECTED_PIPELINE_DATA = {
     "edges": [