From 611ffd13a0cae5ce76efe1d11c23f796482c888b Mon Sep 17 00:00:00 2001
From: sahand_asgarpour <sahand.asgarpour@deltares.nl>
Date: Mon, 4 Dec 2023 08:08:19 +0100
Subject: [PATCH] fix: LineString is impoted from shapely.geometry

---
 ra2ce/graph/origins_destinations.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ra2ce/graph/origins_destinations.py b/ra2ce/graph/origins_destinations.py
index 4b4692f76..bf8f5bc5f 100644
--- a/ra2ce/graph/origins_destinations.py
+++ b/ra2ce/graph/origins_destinations.py
@@ -33,7 +33,7 @@
 import rasterio
 import rasterio.mask
 import rasterio.transform
-import shapely
+from shapely.geometry import LineString
 from rasterio import Affine
 from rasterio.warp import Resampling, calculate_default_transform, reproject
 from shapely.geometry import Point
@@ -180,8 +180,8 @@ def update_edges_with_new_node(
     node_a: int,
     node_b: int,
     k: int,
-    line_a: shapely.LineString,
-    line_b: shapely.LineString,
+    line_a: LineString,
+    line_b: LineString,
     new_node_id: int,
     graph_crs: pyproj.CRS,
     inverse_vertices_dict: dict,