Use key in dict instead of key in dict.keys()

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Aug 5, 2024 · 74a8980 · 74a8980
1 parent f93aa1d
commit 74a8980
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/utils/batch_size.py b/deepmd/utils/batch_size.py
@@ -220,10 +220,10 @@ def append_to_list(res_list, res):
                 results = append_to_list(results, result)
             else:
                 results = (
-                    {kk: [] for kk in result.keys()} if results is None else results
+                    {kk: [] for kk in result} if results is None else results
                 )
                 results = {
-                    kk: append_to_list(results[kk], result[kk]) for kk in result.keys()
+                    kk: append_to_list(results[kk], result[kk]) for kk in result
                 }
         assert results is not None
         assert returned_dict is not None