1 files changed, 13 insertions, 7 deletions
diff --git a/src/python/example/euclidean_witness_complex_diagram_persistence_from_off_file_example.py b/src/python/example/euclidean_witness_complex_diagram_persistence_from_off_file_example.py
index 7587b732..29076c74 100755
--- a/src/python/example/euclidean_witness_complex_diagram_persistence_from_off_file_example.py
+++ b/src/python/example/euclidean_witness_complex_diagram_persistence_from_off_file_example.py
@@ -1,10 +1,14 @@
 #!/usr/bin/env python
 
-import gudhi
 import argparse
+import errno
+import os
+import gudhi
 
-""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
-    See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
+""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ -
+    which is released under MIT.
+    See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full
+    license details.
     Author(s):       Vincent Rouvreau
 
     Copyright (C) 2016 Inria
@@ -45,7 +49,7 @@ with open(args.file, "r") as f:
         print("#####################################################################")
         print("EuclideanWitnessComplex creation from points read in a OFF file")
 
-        witnesses = gudhi.read_off(off_file=args.file)
+        witnesses = gudhi.read_points_from_off_file(off_file=args.file)
         landmarks = gudhi.pick_n_random_points(
             points=witnesses, nb_points=args.number_of_landmarks
         )
@@ -74,9 +78,11 @@ with open(args.file, "r") as f:
         print(simplex_tree.betti_numbers())
 
         if args.no_diagram == False:
-            pplot = gudhi.plot_persistence_diagram(diag, band=args.band)
-            pplot.show()
+            import matplotlib.pyplot as plot
+            gudhi.plot_persistence_diagram(diag, band=args.band)
+            plot.show()
     else:
-        print(args.file, "is not a valid OFF file")
+        raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT),
+                                args.file)
 
     f.close()