Fix WLDataFrame __init__

jnsebgosselin · Mar 2, 2022 · b5c5bad · b5c5bad
1 parent de46459
commit b5c5bad
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gwhat/projet/reader_waterlvl.py b/gwhat/projet/reader_waterlvl.py
@@ -59,7 +59,8 @@
 
 class WLDataFrame(pd.DataFrame):
     def __init__(self, data: list = None, columns: list = None):
-        super().__init__(data=[], columns=COLUMNS, index=[INDEX])
+        super().__init__(data=[], columns=COLUMNS)
+        self.set_index([INDEX], drop=True, inplace=True)
 
         if data is not None and columns is not None:
             df = pd.DataFrame(data, columns=columns)
@@ -143,8 +144,6 @@ def open_water_level_datafile(filename):
                     sheet.iter_rows(min_col=1, values_only=True)]
         finally:
             workbook.close()
-        print(data)
-
     return data