Merge branch 'array-api-learning-rate' of https://github.com/njzjz/deepmd-kit into array-api-learning-rate

Author: njzjz

deepmd/dpmodel/utils/learning_rate.py

@@ -17,7 +17,9 @@
 from deepmd.common import (
     j_get_type,
 )
-from deepmd.dpmodel.array_api import Array
+from deepmd.dpmodel.array_api import (
+    Array,
+)
 from deepmd.utils.plugin import (
     PluginVariant,
     make_plugin_registry,