Issue #6 #8, suppress Future warning at InsertNextPoint2

SciKit-Surgery · Jan 16, 2020 · f201731 · f201731
1 parent 645abad
commit f201731
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sksurgerytrackervisualisation/shapes/dynamic_point_cloud.py b/sksurgerytrackervisualisation/shapes/dynamic_point_cloud.py
@@ -5,7 +5,7 @@
 ability to dynamically add points
 """
 
-from numpy import hstack, ones, int64, arange, ascontiguousarray
+from numpy import hstack, ones, int64, arange, ascontiguousarray, warnings
 from vtk import (vtkPoints, vtkCellArray, vtkPolyData, vtkPolyDataMapper,
                  VTK_ID_TYPE)
 from vtk.util import numpy_support
@@ -69,6 +69,7 @@ def add_point(self, point):
 
         :param: A 3 tuple representing the point coordinate
         """
-
-        self._vtk_points.InsertNextPoint(point)
+        with warnings.catch_warnings(): #see issue #8
+            warnings.simplefilter("ignore", FutureWarning)
+            self._vtk_points.InsertNextPoint(point)
         self._update_actor()