Fix decorator issue (#2542)

NVIDIA · May 1, 2024 · d4b6661 · d4b6661
1 parent 6bd0bf5
commit d4b6661
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nvflare/client/decorator.py b/nvflare/client/decorator.py
@@ -27,7 +27,8 @@ def _replace_func_args(func, kwargs, model: FLModel):
 
 
 class ObjectHolder:
-    pass
+    def __init__(self):
+        self.metrics = None
 
 
 object_holder = ObjectHolder()
@@ -71,7 +72,7 @@ def wrapper(*args, **kwargs):
 
             if object_holder.metrics is not None:
                 return_value.metrics = object_holder.metrics
-                object_holder = None
+                object_holder = ObjectHolder()
 
             send(model=return_value)