Merge pull request #29 from gisce/FIX_extract_fields

Retorna los campos correctos en data_fields
gisce · Jul 12, 2023 · d29a7b3 · d29a7b3
2 parents 10ae9f2 + aea5af1
commit d29a7b3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/powerprofile/powerprofile.py b/powerprofile/powerprofile.py
@@ -439,6 +439,9 @@ def extract(self, cols):
 
             new.curve.rename(columns=cols, inplace=True)
 
+            new_data_fields = [x for x in final_trans_cols if x != self.datetime_field]
+            new.data_fields = new_data_fields
+
         return new
 
     # Dump data