From 77a366420fdd347ce0bb138573fcc7f8268c6012 Mon Sep 17 00:00:00 2001
From: Mario Graff <11542693+mgraffg@users.noreply.github.com>
Date: Thu, 13 Feb 2025 11:51:32 +0000
Subject: [PATCH] wraps regression

---
 CompStats/metrics.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/CompStats/metrics.py b/CompStats/metrics.py
index 3e91f84..cbe04b2 100644
--- a/CompStats/metrics.py
+++ b/CompStats/metrics.py
@@ -318,6 +318,8 @@ def explained_variance_score(y_true,
                              use_tqdm=True,
                              **kwargs):
     """explained_variance_score"""
+
+    @wraps(metrics.explained_variance_score)
     def inner(y, hy):
         return metrics.explained_variance_score(y, hy,
                                                 sample_weight=sample_weight,
@@ -336,6 +338,8 @@ def max_error(y_true, *y_pred,
               use_tqdm=True,
               **kwargs):
     """max_error"""
+
+    @wraps(metrics.max_error)
     def inner(y, hy):
         return metrics.max_error(y, hy)
     return Perf(y_true, *y_pred, score_func=None, error_func=inner,
@@ -354,6 +358,8 @@ def mean_absolute_error(y_true,
                         use_tqdm=True,
                         **kwargs):
     """mean_absolute_error"""
+
+    @wraps(metrics.mean_absolute_error)
     def inner(y, hy):
         return metrics.mean_absolute_error(y, hy,
                                            sample_weight=sample_weight,
@@ -375,6 +381,8 @@ def mean_squared_error(y_true,
                        use_tqdm=True,
                        **kwargs):
     """mean_squared_error"""
+
+    @wraps(metrics.mean_squared_error)
     def inner(y, hy):
         return metrics.mean_squared_error(y, hy,
                                           sample_weight=sample_weight,
@@ -396,6 +404,8 @@ def root_mean_squared_error(y_true,
                             use_tqdm=True,
                             **kwargs):
     """root_mean_squared_error"""
+
+    @wraps(metrics.root_mean_squared_error)
     def inner(y, hy):
         return metrics.root_mean_squared_error(y, hy,
                                                sample_weight=sample_weight,
@@ -417,6 +427,8 @@ def mean_squared_log_error(y_true,
                            use_tqdm=True,
                            **kwargs):
     """mean_squared_log_error"""
+
+    @wraps(metrics.mean_squared_log_error)
     def inner(y, hy):
         return metrics.mean_squared_log_error(y, hy,
                                               sample_weight=sample_weight,
@@ -438,6 +450,8 @@ def root_mean_squared_log_error(y_true,
                                 use_tqdm=True,
                                 **kwargs):
     """root_mean_squared_log_error"""
+
+    @wraps(metrics.root_mean_squared_log_error)
     def inner(y, hy):
         return metrics.root_mean_squared_log_error(y, hy,
                                                    sample_weight=sample_weight,
@@ -459,6 +473,8 @@ def median_absolute_error(y_true,
                           use_tqdm=True,
                           **kwargs):
     """median_absolute_error"""
+
+    @wraps(metrics.median_absolute_error)
     def inner(y, hy):
         return metrics.median_absolute_error(y, hy,
                                              sample_weight=sample_weight,
@@ -481,6 +497,8 @@ def r2_score(y_true,
              use_tqdm=True,
              **kwargs):
     """r2_score"""
+
+    @wraps(metrics.r2_score)
     def inner(y, hy):
         return metrics.r2_score(y, hy,
                                 sample_weight=sample_weight,
@@ -502,6 +520,8 @@ def mean_poisson_deviance(y_true,
                           use_tqdm=True,
                           **kwargs):
     """mean_poisson_deviance"""
+
+    @wraps(metrics.mean_poisson_deviance)
     def inner(y, hy):
         return metrics.mean_poisson_deviance(y, hy,
                                              sample_weight=sample_weight)
@@ -521,6 +541,8 @@ def mean_gamma_deviance(y_true,
                         use_tqdm=True,
                         **kwargs):
     """mean_gamma_deviance"""
+
+    @wraps(metrics.mean_gamma_deviance)
     def inner(y, hy):
         return metrics.mean_gamma_deviance(y, hy,
                                            sample_weight=sample_weight)
@@ -541,6 +563,8 @@ def mean_absolute_percentage_error(y_true,
                                    use_tqdm=True,
                                    **kwargs):
     """mean_absolute_percentage_error"""
+
+    @wraps(metrics.mean_absolute_percentage_error)
     def inner(y, hy):
         return metrics.mean_absolute_percentage_error(y, hy,
                                                       sample_weight=sample_weight,
@@ -562,6 +586,8 @@ def d2_absolute_error_score(y_true,
                             use_tqdm=True,
                             **kwargs):
     """d2_absolute_error_score"""
+
+    @wraps(metrics.d2_absolute_error_score)
     def inner(y, hy):
         return metrics.d2_absolute_error_score(y, hy,
                                                sample_weight=sample_weight,