fix models performance

Nixtla · Nov 29, 2024 · 41b0d21 · 41b0d21
1 parent 78d01ec
commit 41b0d21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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()}