add stacklevel = 1 to warnings.warn in ElasticNetRegression

Safe-DS · Apr 21, 2023 · 1cd2d42 · 1cd2d42
1 parent 34f4562
commit 1cd2d42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/safeds/ml/classical/regression/_elastic_net_regression.py b/src/safeds/ml/classical/regression/_elastic_net_regression.py
@@ -21,10 +21,10 @@ def __init__(self, lasso_ratio: float = .5) -> None:
             raise ValueError("lasso_ratio must be between 0 and 1.")
         elif lasso_ratio == 0:
             warnings.warn("ElasticNetRegression with lasso_ratio = 0 is essentially RidgeRegression."
-                          " Use RidgeRegression instead for better numerical stability.")
+                          " Use RidgeRegression instead for better numerical stability.", stacklevel=1)
         elif lasso_ratio == 1:
             warnings.warn("ElasticNetRegression with lasso_ratio = 0 is essentially LassoRegression."
-                          " Use LassoRegression instead for better numerical stability.")
+                          " Use LassoRegression instead for better numerical stability.", stacklevel=1)
         self.lasso_ratio = lasso_ratio
         self._wrapped_regressor: sk_ElasticNet | None = None
         self._feature_names: list[str] | None = None