From 41b0d21d7a95c5bc3dfc2468a236709cfaae03cd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Fri, 29 Nov 2024 12:08:56 -0600
Subject: [PATCH] fix models performance

---
 .pre-commit-config.yaml                 | 3 ++-
 action_files/models_performance/main.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 4c7499d9..8b9d930e 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -17,10 +17,11 @@ repos:
     rev: v0.2.1
     hooks:
       - id: ruff
+        files: 'nixtla'
 
   - repo: https://github.com/pre-commit/mirrors-mypy
     rev: v1.10.1
     hooks:
       - id: mypy
         args: [--ignore-missing-imports]
-        exclude: "setup.py"
+        files: 'nixtla'
diff --git a/action_files/models_performance/main.py b/action_files/models_performance/main.py
index 22a631a5..0d20d995 100644
--- a/action_files/models_performance/main.py
+++ b/action_files/models_performance/main.py
@@ -184,7 +184,7 @@ def evaluate_benchmark_performace(self) -> Tuple[pd.DataFrame, pd.DataFrame]:
                 h=self.h,
                 n_windows=self.n_windows,
                 step_size=self.h,
-            ).reset_index()
+            )
             total_time = time() - init_time
             cv_model_df = cv_model_df.rename(
                 columns={value: key for key, value in renamer.items()}