fix: raise error if empty header

daavoo · Aug 8, 2023 · 001ce2c · 001ce2c
1 parent 4be3d13
commit 001ce2c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyntcloud/io/off.py b/pyntcloud/io/off.py
@@ -12,6 +12,7 @@ def read_off(filename):
         if "OFF" not in first_line:
             raise ValueError("The file does not start with the word OFF")
         has_color = "C" in first_line
+        num_rows = None
 
         # Read header.
         for line in f:
@@ -25,6 +26,9 @@ def read_off(filename):
             num_rows = n_points + n_faces
             break
 
+        if num_rows is None:
+            raise ValueError("The file does not contain a valid header")
+
         # Read remaining lines.
         lines = [next(f) for _ in range(num_rows)]