simplify vtk import

plotly · Feb 4, 2021 · a7fc126 · a7fc126
1 parent e3ecc95
commit a7fc126
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dash_vtk/utils/vtk.py b/dash_vtk/utils/vtk.py
@@ -1,7 +1,7 @@
 import dash_vtk
 
-import vtk
 from vtk.util.numpy_support import vtk_to_numpy
+from vtk.vtkFiltersGeometry import vtkGeometryFilter
 
 # Numpy to JS TypedArray
 to_js_type = {
@@ -28,7 +28,7 @@ def to_mesh_state(dataset, field_to_keep=None):
   if dataset.IsA('vtkPolyData'):
     polydata = dataset
   else:
-    extractSkinFilter = vtk.vtkGeometryFilter()
+    extractSkinFilter = vtkGeometryFilter()
     extractSkinFilter.SetInputData(dataset)
     extractSkinFilter.Update()
     polydata = extractSkinFilter.GetOutput()