Update spatial.py

pyswmm · Sep 25, 2023 · 99172bc · 99172bc
1 parent 90cfcfe
commit 99172bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/swmmio/utils/spatial.py b/swmmio/utils/spatial.py
@@ -216,6 +216,8 @@ def write_shapefile(df, filename, geomtype='line', prj=None):
             w.line([row.coords])
         if geomtype == 'point':
             w.point(*row.coords[0])
+        if geomtype== 'polygon': 
+            w.poly([row.coords])
 
     w.close()