add geometry docstring to model/base (tardis-sn#102)

smokestacklightnin · Jul 25, 2023 · 306c451 · 306c451
1 parent 5d72332
commit 306c451
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stardis/model/base.py b/stardis/model/base.py
@@ -14,6 +14,7 @@ class StellarModel:
     fv_geometry : pandas.core.frame.DataFrame
     abundances : pandas.core.frame.DataFrame
     boundary_temps : numpy.ndarray
+    geometry : stardis.model.geometry
 
     Attributes
     ----------
@@ -23,6 +24,8 @@ class StellarModel:
         Abundance DataFrame with all included elements and mass abundances.
     boundary_temps : numpy.ndarray
         Temperatures in K of all shell boundaries. Note that array is transposed.
+    geometry : stardis.model.geometry
+        Geometry of the model.
     """
 
     def __init__(self, fv_geometry, abundances, boundary_temps, geometry):