import vtk in prediction.py

baptistebaquero · Apr 7, 2022 · dbae693 · dbae693
1 parent e997eb9
commit dbae693
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/py/prediction.py b/py/prediction.py
@@ -12,6 +12,7 @@
 from monai.transforms import AsDiscrete
 from monai.data import decollate_batch
 import shutil
+import vtk
 from scipy import linalg
 
 def main(args):