33# SPDX-FileContributor: Lionel Untereiner
44from typing_extensions import Self
55
6- import vtkmodules .all as vtk
6+ from vtkmodules .vtkCommonCore import vtkInformation ,vtkInformationVector
7+ from vtkmodules .vtkCommonDataModel import vtkPartitionedDataSetCollection
78from vtkmodules .util .vtkAlgorithm import VTKPythonAlgorithmBase
89
910from geos_xml_viewer .algorithms .deck import SimulationDeck , build_model , read
@@ -53,9 +54,9 @@ def GetAttributeName( self: Self ) -> str:
5354
5455 def RequestData (
5556 self : Self ,
56- request : vtk . vtkInformation ,
57- inInfoVec : vtk . vtkInformationVector ,
58- outInfoVec : vtk . vtkInformationVector ,
57+ request : vtkInformation ,
58+ inInfoVec : vtkInformationVector ,
59+ outInfoVec : vtkInformationVector ,
5960 ) -> int :
6061 """RequestData function of the vtk pipeline.
6162
@@ -68,9 +69,9 @@ def RequestData(
6869 int: Returns 1 if the pipeline is successful
6970 """
7071 self .__simulationDeck = read ( self .__filename )
71- opt = vtk . vtkPartitionedDataSetCollection .GetData ( outInfoVec )
72+ opt = vtkPartitionedDataSetCollection .GetData ( outInfoVec )
7273
73- output = vtk . vtkPartitionedDataSetCollection ()
74+ output = vtkPartitionedDataSetCollection ()
7475 build_model ( self .__simulationDeck , output , self .__attributeName )
7576
7677 opt .ShallowCopy ( output )
0 commit comments