Update metalearners/metalearner_grid_search_cv.py

Co-authored-by: Kevin Klein <[email protected]>
Quantco · Jun 27, 2024 · a14932c · a14932c
1 parent d4cfb2a
commit a14932c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metalearners/metalearner_grid_search_cv.py b/metalearners/metalearner_grid_search_cv.py
@@ -75,7 +75,7 @@ def _fit_and_score(job: _FitAndScoreJob) -> _CVResult:
     )
 
 
-def _format_results(results: Sequence[_CVResult]):
+def _format_results(results: Sequence[_CVResult]) -> pd.DataFrame:
     rows = []
     for result in results:
         row: dict[str, str | int | float] = {}