diff --git a/main.py b/main.py index 6e3e8e1..295e8d3 100644 --- a/main.py +++ b/main.py @@ -105,9 +105,10 @@ def main(): def reduce_columns(df, filename): if '15MIN' in filename: - return df.drop(columns=['Month', 'Year', 'date']+weak_column_names) + return df.drop(columns=['Month', 'Year', 'date', 'DayOfWeek']+weak_column_names, errors='ignore') else: - return df.drop(columns=['Month', 'Year', 'date']) + return df.drop(columns=['Month', 'Year', 'date', 'DayOfWeek'], errors='ignore') + def main_two(): results = pd.DataFrame()