Added stack level 2 to the warnings

next-exp · Apr 6, 2021 · 58e1993 · 58e1993
1 parent 4959384
commit 58e1993
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/invisible_cities/cities/components.py b/invisible_cities/cities/components.py
@@ -131,11 +131,12 @@ def create_timestamp(rate: float) -> float:
 
     if rate == 0:
         warnings.warn("Zero rate is unphysical, using default "
-                      "rate = 0.5 Hz instead")
+                      "rate = 0.5 Hz instead", stacklevel=2)
         rate = 0.5 * units.hertz
     elif rate < 0:
         warnings.warn(f"Negative rate is unphysical, using "
-                      f"rate = {abs(rate) / units.hertz} Hz instead")
+                      f"rate = {abs(rate) / units.hertz} Hz instead",
+                      stacklevel=2)
         rate = abs(rate)
 
     def create_timestamp_(event_number: Union[int, float]) -> float: