diff --git a/scripts/openmc-track-to-vtk b/scripts/openmc-track-to-vtk index 1761d08ba6..8db57e2a3f 100755 --- a/scripts/openmc-track-to-vtk +++ b/scripts/openmc-track-to-vtk @@ -89,7 +89,10 @@ def main(): data.SetLines(cells) writer = vtk.vtkXMLPPolyDataWriter() - writer.SetInputData(data) + if vtk.vtkVersion.GetVTKMajorVersion() > 5: + writer.SetInputData(data) + else: + writer.SetInput(data) writer.SetFileName(args.out) writer.Write() diff --git a/scripts/openmc-voxel-to-silovtk b/scripts/openmc-voxel-to-silovtk index 633d44cb12..eb052c75fa 100755 --- a/scripts/openmc-voxel-to-silovtk +++ b/scripts/openmc-voxel-to-silovtk @@ -62,7 +62,10 @@ def main(file_, o): grid.GetCellData().AddArray(data) writer = vtk.vtkXMLImageDataWriter() - writer.SetInputData(grid) + if vtk.vtkVersion.GetVTKMajorVersion() > 5: + writer.SetInputData(grid) + else: + writer.SetInput(grid) if not o.output.endswith(".vti"): o.output += ".vti" writer.SetFileName(o.output)