fix pylint

InternLM · Sep 11, 2024 · 71aa031 · 71aa031
1 parent 24e391a
commit 71aa031
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internlm/data/megatron/dataset.py b/internlm/data/megatron/dataset.py
@@ -30,8 +30,8 @@ def print_rank_0(message):
 
 
 def code(dtype):
-    for k in dtypes.keys():
-        if dtypes[k] == dtype:
+    for k, v in dtypes.items():
+        if v == dtype:
             return k
     raise ValueError(dtype)