Skip to content

Commit 4ff0877

Browse files
authored
Merge pull request #151 from Ipuch/pyorerun
fix: extra viz mass check
2 parents 3c44753 + 46b6478 commit 4ff0877

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

bionc/vizualization/pyorerun_interface.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def segments(self) -> tuple["BioncSegment", ...]:
5353

5454
@property
5555
def segment_names_with_mass(self) -> tuple[str, ...]:
56-
return tuple([segment.name for segment in self.model.segments.values() if segment.mass > 0])
56+
return tuple(
57+
[segment.name for segment in self.model.segments.values() if segment.mass is not None and segment.mass > 0]
58+
)
5759

5860
def segment_homogeneous_matrices_in_global(self, q: np.ndarray, segment_index: int) -> np.ndarray:
5961
"""

0 commit comments

Comments
 (0)