refactor: move group_map to builder

Author: aignas

python/private/pypi/extension.bzl

@@ -538,12 +538,6 @@ You cannot use both the additive_build_content and additive_build_content_file a
                     intersection[pkg] = None
                 exposed_packages[hub_name] = intersection
 
-            # TODO @aignas 2024-04-05: how do we support different requirement
-            # cycles for different abis/oses? For now we will need the users to
-            # assume the same groups across all versions/platforms until we start
-            # using an alternative cycle resolution strategy.
-            hub_group_map[hub_name] = pip_attr.experimental_requirement_cycles
-
     for hub in pip_hub_map.values():
         hub_whl_map.setdefault(hub.name, {})
         for key, settings in hub.whl_map.items():
@@ -558,6 +552,8 @@ You cannot use both the additive_build_content and additive_build_content_file a
                 # replicate whl_libraries.update(out.whl_libraries)
                 whl_libraries[whl_name] = lib
 
+        hub_group_map[hub.name] = hub.group_map
+
     return struct(
         # We sort so that the lock-file remains the same no matter the order of how the
         # args are manipulated in the code going before.

python/private/pypi/hub_builder.bzl

@@ -47,6 +47,7 @@ def hub_builder(
         exposed_packages = {},
         extra_aliases = {},
         whl_libraries = {},
+        group_map = {},
         _evaluate_markers_fn = evaluate_markers_fn,
         _logger = logger,
         _minor_mapping = minor_mapping,
@@ -93,6 +94,13 @@ def _add(self, *, pip_attr):
     _set_index_urls(self, pip_attr)
     self._pip_attrs[python_version] = pip_attr
 
+    # TODO @aignas 2024-04-05: how do we support different requirement
+    # cycles for different abis/oses? For now we will need the users to
+    # assume the same groups across all versions/platforms until we start
+    # using an alternative cycle resolution strategy.
+    self.group_map.clear()
+    self.group_map.update(pip_attr.experimental_requirement_cycles)
+
 def _set_index_urls(self, pip_attr):
     if not pip_attr.experimental_index_url:
         if pip_attr.experimental_extra_index_urls: