fix dataset for python38 (#2065)

unit8co · Nov 11, 2023 · d206055 · d206055
1 parent 09300d9
commit d206055
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/darts/datasets/__init__.py b/darts/datasets/__init__.py
@@ -858,8 +858,8 @@ def pre_process_dataset(dataset_path):
             """Restrict the time axis and add the weather data"""
             df = pd.read_csv(dataset_path, index_col=0)
             # convert time index
-            df.index = (
-                pd.DatetimeIndex(df.index, tz="CET").tz_convert("UTC").tz_localize(None)
+            df.index = pd.DatetimeIndex(pd.to_datetime(df.index, utc=True)).tz_localize(
+                None
             )
             # extract pre-determined period
             df = df.loc[
@@ -914,8 +914,8 @@ def _download_weather_data():
         df["param_name"] = df["Parameter"] + " [" + df["Einheit"] + "]"
         df = df.pivot(index="Datum", columns="param_name", values="Wert")
         # convert time index to from CET to UTC and extract the required time range
-        df.index = (
-            pd.DatetimeIndex(df.index, tz="CET").tz_convert("UTC").tz_localize(None)
+        df.index = pd.DatetimeIndex(pd.to_datetime(df.index, utc=True)).tz_localize(
+            None
         )
         df = df.loc[
             (pd.Timestamp("2015-01-01") <= df.index)