Enhance previous commit code

AlexeyPechnikov · Mar 16, 2024 · dcefc79 · dcefc79
1 parent d8bbdba
commit dcefc79
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygmtsar/pygmtsar/Stack_export.py b/pygmtsar/pygmtsar/Stack_export.py
@@ -252,7 +252,7 @@ def export_vtk(self, data, name, caption='Exporting WGS84 VTK(s)', topo='auto',
         """
         import xarray as xr
         import numpy as np
-        from vtk import vtkStructuredGridWriter, vtkStringArray
+        from vtk import vtkStructuredGridWriter, vtkStringArray, VTK_BINARY
         from tqdm.auto import tqdm
         import os
 
@@ -321,6 +321,6 @@ def export_vtk(self, data, name, caption='Exporting WGS84 VTK(s)', topo='auto',
                 writer = vtkStructuredGridWriter()
                 writer.SetFileName(filename)
                 writer.SetInputData(vtk_grid)
-                writer.SetFileType(vtk.VTK_BINARY)
+                writer.SetFileType(VTK_BINARY)
                 writer.Write()
                 pbar.update(1)