lint: don't import too much from root

We'd better import NWR from yuheng.type
geo-yuheng · Feb 28, 2024 · 79eec19 · 79eec19
1 parent 62a8719
commit 79eec19
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/yuheng/plugin/driver_db_postgresql/__main__.py b/src/yuheng/plugin/driver_db_postgresql/__main__.py
@@ -11,7 +11,8 @@
 current_dir = os.path.dirname(os.path.realpath(__file__))
 src_dir = os.path.join(current_dir, "..", "..", "..")
 sys.path.append(src_dir)
-from yuheng import Node, Waifu, Way
+from yuheng import Waifu
+from yuheng.type import Node, Way
 
 PROJ_TRANSFORMER = pyproj.Transformer.from_crs("epsg:3857", "epsg:4326")
 

diff --git a/tests/cases/plugin_folium.py b/tests/cases/plugin_folium.py
@@ -6,7 +6,8 @@
 src_dir = os.path.join(current_dir, "../../src")
 sys.path.append(src_dir)
 
-from yuheng import Member, Node, Relation, Waifu, Way
+from yuheng import Waifu
+from yuheng.type import Member, Node, Relation, Way
 from yuheng.plugin.folium.__main__ import display