parametrization: move comment, fix exception error msg (#5142)

Author: skshetry

dvc/parsing/__init__.py

@@ -136,14 +136,14 @@ def __init__(self, repo: "Repo", wdir: PathInfo, d: dict):
         except ContextError as exc:
             format_and_raise(exc, "'vars'", self.relpath)
 
-        # we wrap the definitions into ForeachDefinition and EntryDefinition,
-        # that helps us to optimize, cache and selectively load each one of
-        # them as we need, and simplify all of this DSL/parsing logic.
         # we use `tracked_vars` to keep a dictionary of used variables
         # by the interpolated entries.
         self.tracked_vars: Dict[str, Mapping] = {}
 
         stages_data = d.get(STAGES_KWD, {})
+        # we wrap the definitions into ForeachDefinition and EntryDefinition,
+        # that helps us to optimize, cache and selectively load each one of
+        # them as we need, and simplify all of this DSL/parsing logic.
         self.definitions: Dict[str, Definition] = {
             name: make_definition(self, name, definition)
             for name, definition in stages_data.items()

dvc/parsing/context.py

@@ -335,9 +335,9 @@ def select(
                     Defaults to False. Note that the default is different from
                     `resolve`.
         """
-        key = normalize_key(key)
+        normalized = normalize_key(key)
         try:
-            node = super().select(key)
+            node = super().select(normalized)
         except ValueError as exc:
             raise KeyNotInContext(key) from exc