From ae79c8808785144d7a665ba3bdf314e1b61c195b Mon Sep 17 00:00:00 2001
From: Bane Sullivan <banesullivan@gmail.com>
Date: Mon, 1 Jan 2024 13:07:21 -0800
Subject: [PATCH] Fix points_to_polydata

---
 PVGeo/interface.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/PVGeo/interface.py b/PVGeo/interface.py
index 21ea7318..558099d7 100644
--- a/PVGeo/interface.py
+++ b/PVGeo/interface.py
@@ -179,16 +179,11 @@ def points_to_poly_data(points, copy_z=False):
     pdata = pv.PolyData(points)
 
     # Add attributes if given
-    scalSet = False
     for i, key in enumerate(keys):
-        data = convert_array(atts[:, i], name=key)
-        pdata.GetPointData().AddArray(data)
-        if not scalSet:
-            pdata.GetPointData().SetActiveScalars(key)
-            scalSet = True
+        pdata.point_data[key] = convert_array(atts[:, i])
     if copy_z:
         z = convert_array(points[:, 2], name='Elevation')
-        pdata.GetPointData().AddArray(z)
+        pdata.point_data['Elevation'] = z
     return pv.wrap(pdata)