some small fixes

tardis-sn · Oct 30, 2015 · 730d2c8 · 730d2c8
1 parent 5400100
commit 730d2c8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tardis/simulation/base.py b/tardis/simulation/base.py
@@ -10,9 +10,6 @@
 
 from astropy import units as u
 
-from collections import OrderedDict
-
-from tardis.model import Radial1DModel
 from tardis.montecarlo.base import MontecarloRunner
 
 # Adding logging support
@@ -145,7 +142,8 @@ def log_plasma_state(self, t_rad, w, t_inner, next_t_rad, next_w,
             next t_rad
         next_w: ~astropy.units.Quanity
             next_w
-        log_sampling
+        log_sampling: ~int
+            the n-th shells to be plotted
 
         Returns
         -------