Update deepmd/dpmodel/utils/serialization.py

Signed-off-by: Jinzhe Zeng <[email protected]>

Author: njzjz

deepmd/dpmodel/utils/serialization.py

@@ -234,7 +234,7 @@ def size(self) -> int:
         """
         total_size = 0
 
-        def count_variables(x: Any) -> None:
+        def count_variables(x: Any) -> Any:
             nonlocal total_size
             if isinstance(x, np.ndarray):
                 total_size += x.size