docs: cleaned up docstrings

Author: HarrisonTCodes

src/ursina_proteins/atoms.py

@@ -75,15 +75,15 @@ def __init__(
         **kwargs,
     ):
         """
-        Create an entity to visualise atoms
+        Create an entity to visualise atoms.
 
         Args:
             atoms: Atoms to visualise.
-            atom_size: Size of individual atoms in the atoms mesh.
+            atom_size: Size of individual atoms.
             atom_vertices: Base vertices to use for atom geometry (default: None).
             atom_triangles: Base triangles to use for atom geometry (default: None).
             atom_normals: Base normals to use for atom geometry (default: None).
-            element_color_map: Color mapping for atoms (default: None).
+            element_color_map: Color mapping for atom elements (default: None).
             *args: Arguments passed to constructor for the entity.
             **kwargs: Keyword arguments passed to constructor for the entity.
         """

src/ursina_proteins/chains.py

@@ -38,7 +38,7 @@ def __init__(
         **kwargs,
     ):
         """
-        Create an entity to visualise helices and a subentity to visualise coils of chains
+        Create an entity to visualise helices and a subentity to visualise coils of chains.
 
         Args:
             chains: Chains to visualise.

src/ursina_proteins/protein.py

@@ -48,7 +48,7 @@ def __init__(
         **kwargs,
     ):
         """
-        Parse a protein structure file and create visualisation entities
+        Parse a protein structure file and create visualisation entities.
 
         Args:
             protein_filepath: Path to the protein file.
@@ -57,13 +57,13 @@ def __init__(
             chains_smoothness: Smoothness factor for chain rendering (default: 3).
             helices_thickness: Thickness of helix meshes (default: 4).
             coils_thickness: Thickness of coil meshes (default: 1).
-            chain_id_color_map: Color mapping for chains (default: None).
+            chain_id_color_map: Color mapping for chain IDs (default: None).
             compute_atoms: Flag to enable/disable atoms computation (default: True).
             atom_size: Size of individual atoms in the atoms mesh (default: 0.1).
             atom_vertices: Base vertices to use for atom geometry (default: None).
             atom_triangles: Base triangles to use for atom geometry (default: None).
             atom_normals: Base normals to use for atom geometry (default: None).
-            atom_element_color_map: Color mapping for atoms (default: None).
+            atom_element_color_map: Color mapping for atom elements (default: None).
             *args: Arguments passed to constructor for each entity.
             **kwargs: Keyword arguments passed to constructor for each entity.
         """