Fixed small bug

hangg7 · Jun 6, 2023 · adfd1de · adfd1de
1 parent 259f6cb
commit adfd1de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/nerfvis/scene.py b/nerfvis/scene.py
@@ -415,8 +415,8 @@ def add_mesh(self, name : str,
             self.fields[_f(name, "face_size")] = int(face_size)
             assert face_size >= 1 and face_size <= 3
         if faces is not None:
-            assert faces.ndim == 2 and faces.shape[1] == face_size, \
-                    f"faces must be (N, face_size={face_size})"
+            assert faces.ndim == 2 and (face_size is None or faces.shape[1] == face_size), \
+                    f"faces must be (N, face_size={face_size if face_size is not None else -1})"
         if faces is not None:
             self.fields[_f(name, "faces")] = _to_np_array(faces).astype(np.int32)
         self._update_bb(points, **kwargs)

diff --git a/nerfvis/version.py b/nerfvis/version.py
@@ -1 +1 @@
-__version__ = '0.1.6'
+__version__ = '0.1.7'