From b5c5bad495ef5791f93f541164841619dcd038e7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jean-S=C3=A9bastien=20Gosselin?= Date: Wed, 2 Mar 2022 09:58:12 -0500 Subject: [PATCH] Fix WLDataFrame __init__ --- gwhat/projet/reader_waterlvl.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/gwhat/projet/reader_waterlvl.py b/gwhat/projet/reader_waterlvl.py index 14ab67f7..2eb0bf80 100644 --- 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