Fix a typo

crsky1023 · Feb 8, 2022 · 92f6d3f · 92f6d3f
1 parent f1fe8f8
commit 92f6d3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/muon/_core/utils.py b/muon/_core/utils.py
@@ -118,7 +118,7 @@ def _get_values(
         if (use_raw is None or use_raw) and layer is not None:
             for m in data.mod:
                 if key_in_mod[m] == False and data.mod[m].raw is not None:
-                    key_in_mod[m] = k in data.mod[m].raw.var_names
+                    key_in_mod[m] = key in data.mod[m].raw.var_names
                     if key_in_mod[m] and data.mod[m].raw is None and layer is None:
                         warnings.warn(
                             f"Attibute .raw is None for the modality {m}, using .X instead"