Bug fix for indices.py function pe

timcera · Aug 16, 2023 · c8bea7c · c8bea7c
1 parent 9e5ce06
commit c8bea7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mettoolbox/indices.py b/src/mettoolbox/indices.py
@@ -130,9 +130,9 @@ def pe(
         target_units=["mm", "mm"],
     )
 
-    pe_data = tsd["rainfall:mm"] - tsd["pet:mm"]
+    tsd["pe:mm"] = tsd["rainfall:mm"] - tsd["pet:mm"]
 
-    pe_data = tsutils._normalize_units(pe_data, "mm", target_units)
+    pe_data = tsutils._normalize_units(tsd["pe:mm"], "mm", target_units)
 
     pe_data = (
         pe_data.astype(float)