diff --git a/examples/benchmarks/LightGBM/features_sample.py b/examples/benchmarks/LightGBM/features_sample.py index e06637b84..72fffb79d 100644 --- a/examples/benchmarks/LightGBM/features_sample.py +++ b/examples/benchmarks/LightGBM/features_sample.py @@ -11,6 +11,7 @@ class ResampleProcessor(InstProcessor): self.minute = minute def __call__(self, df: pd.DataFrame, *args, **kwargs): + df.index = pd.to_datetime(df.index) df = df.loc[df.index.time == datetime.time(self.hour, self.minute)] df.index = df.index.normalize() return df