fix stacklevel error

adap · Dec 21, 2023 · 437de39 · 437de39
1 parent bbeb1a5
commit 437de39
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/py/flwr/server/strategy/dpfedavg_adaptive.py b/src/py/flwr/server/strategy/dpfedavg_adaptive.py
@@ -35,10 +35,11 @@ class DPFedAvgAdaptive(DPFedAvgFixed):
     warnings.warn(
         "DPFedAvgAdaptive is deprecated and will be removed in future versions.",
         DeprecationWarning,
+        stacklevel=2,
     )
     """Wrapper for configuring a Strategy for DP with Adaptive Clipping."""
 
-    # pylint: disable=too-many-arguments,too-many-instance-attributes
+    # pylint: disable=too-many-arguments,too-many-instance-attributes,line-too-long,missing-docstring
     def __init__(
         self,
         strategy: Strategy,
@@ -53,6 +54,7 @@ def __init__(
         warnings.warn(
             "DPFedAvgAdaptive is deprecated and will be removed in future versions.",
             DeprecationWarning,
+            stacklevel=2,
         )
 
         super().__init__(

diff --git a/src/py/flwr/server/strategy/dpfedavg_fixed.py b/src/py/flwr/server/strategy/dpfedavg_fixed.py
@@ -33,10 +33,11 @@ class DPFedAvgFixed(Strategy):
     warnings.warn(
         "DPFedAvgFixed is deprecated and will be removed in future versions.",
         DeprecationWarning,
+        stacklevel=2,
     )
     """Wrapper for configuring a Strategy for DP with Fixed Clipping."""
 
-    # pylint: disable=too-many-arguments,too-many-instance-attributes
+    # pylint: disable=too-many-arguments,too-many-instance-attributes, line-too-long
     def __init__(
         self,
         strategy: Strategy,
@@ -48,6 +49,7 @@ def __init__(
         warnings.warn(
             "DPFedAvgFixed is deprecated and will be removed in future versions.",
             DeprecationWarning,
+            stacklevel=2,
         )
 
         super().__init__()