move name normalization to parse_requirements

Author: aignas

python/private/pypi/extension.bzl

@@ -218,8 +218,6 @@ def _create_whl_repos(
     )
 
     for whl_name, requirements in requirements_by_platform.items():
-        whl_name = normalize_name(whl_name)
-
         group_name = whl_group_mapping.get(whl_name)
         group_deps = requirement_cycles.get(group_name, [])
 
@@ -333,7 +331,7 @@ def _create_whl_repos(
         is_reproducible = is_reproducible,
         whl_map = whl_map,
         exposed_packages = {
-            normalize_name(whl_name): None
+            whl_name: None
             for whl_name, requirements in requirements_by_platform.items()
             if len([r for r in requirements if r.is_exposed]) > 0
         },

python/private/pypi/parse_requirements.bzl

@@ -203,6 +203,9 @@ def parse_requirements(
                 sorted(requirements),
             ))
 
+        # Return normalized names
+        ret_requirements = ret.setdefault(normalize_name(whl_name), [])
+
         for r in sorted(reqs.values(), key = lambda r: r.requirement_line):
             whls, sdist = _add_dists(
                 requirement = r,
@@ -211,7 +214,7 @@ def parse_requirements(
             )
 
             target_platforms = env_marker_target_platforms.get(r.requirement_line, r.target_platforms)
-            ret.setdefault(whl_name, []).append(
+            ret_requirements.append(
                 struct(
                     distribution = r.distribution,
                     srcs = r.srcs,