fix: fixed handling of chains without helices

HarrisonTCodes · HarrisonTCodes · commit 3716dff33485 · 2025-10-14T17:23:17.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "ursina-proteins"
-version = "1.5.2"
+version = "1.5.3"
 description = "A Python package for rendering protein structures of PDB format in 3D using Ursina"
 authors = [{ name = "HarrisonTCodes" }]
 requires-python = ">=3.12"
diff --git a/src/ursina_proteins/protein.py b/src/ursina_proteins/protein.py
@@ -298,8 +298,9 @@ def compute_helices_and_coils_meshes(
 
             # Chain info
             chain_id = chain.get_id()
+            chain_helices = self.helices.get(chain_id) or []
             chain_segments = parse_segments(
-                self.helices[chain_id], len(carbon_alpha_coords), "helices", "coils"
+                chain_helices, len(carbon_alpha_coords), "helices", "coils"
             )
 
             # Render each segment (helices and coils)
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -298,8 +298,9 @@ def compute_helices_and_coils_meshes(`
`298`	`298`
`299`	`299`	`# Chain info`
`300`	`300`	`chain_id = chain.get_id()`
	`301`	`+ chain_helices = self.helices.get(chain_id) or []`
`301`	`302`	`chain_segments = parse_segments(`
`302`		`- self.helices[chain_id], len(carbon_alpha_coords), "helices", "coils"`
	`303`	`+ chain_helices, len(carbon_alpha_coords), "helices", "coils"`
`303`	`304`	`)`
`304`	`305`
`305`	`306`	`# Render each segment (helices and coils)`