From 84fad566cc699a5ac2ad4f647444ef67eb93d60b Mon Sep 17 00:00:00 2001
From: CaRoLZhangxy <ucaszxy@outlook.com>
Date: Mon, 11 Mar 2024 07:46:16 +0000
Subject: [PATCH] change code style

---
 deepmd/pt/optimizer/LKF.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/deepmd/pt/optimizer/LKF.py b/deepmd/pt/optimizer/LKF.py
index 069dcea1fc..2dc934263c 100644
--- a/deepmd/pt/optimizer/LKF.py
+++ b/deepmd/pt/optimizer/LKF.py
@@ -32,13 +32,13 @@ def __init__(
         kalman_nue=0.9987,
         block_size=5120,
     ):
-        defaults = dict(
-            lr=0.1,
-            kalman_nue=kalman_nue,
-            block_size=block_size,
-        )
+        defaults = {
+            "lr":0.1,
+            "kalman_nue":kalman_nue,
+            "block_size":block_size
+        }
 
-        super(LKFOptimizer, self).__init__(params, defaults)
+        super().__init__(params, defaults)
 
         self._params = self.param_groups[0]["params"]