pylint flake8 fixes

Author: colourmeamused

dea_conflux/stack.py

@@ -194,8 +194,12 @@ def remove_timeseries_with_duplicated(df: pd.DataFrame) -> pd.DataFrame:
     # The pc_missing the less the better, so we only keep the first one
     df = df.drop_duplicates("DAY", keep="first")
 
-    # Remove entries within 60s, this takes care of the edge case where duplicates wrap across midnight UTC
-    df['TIMEDIFF'] = pd.to_datetime(df['date'].shift(-1)) - pd.to_datetime(df['date'])
+    # Remove entries within 60s, deals with edge cases where duplicates wrap across midnight UTC
+
+    if "date" not in df.columns:
+        df['TIMEDIFF'] = pd.to_datetime(df.index.shift(-1)) - pd.to_datetime(df.index)
+    else:
+        df['TIMEDIFF'] = pd.to_datetime(df['date'].shift(-1)) - pd.to_datetime(df['date'])
     df = df[~(df['TIMEDIFF'] < timedelta(seconds=60))]
     df = df.drop(columns=["TIMEDIFF"])