fixup tests

Author: aignas

tests/pypi/hub_builder/hub_builder_tests.bzl

@@ -332,7 +332,9 @@ torch==2.4.1 ; platform_machine != 'x86_64' \
         _parse(
             hub_name = "pypi",
             python_version = "3.15",
-            requirements_lock = "universal.txt",
+            requirements_by_platform = {
+                "universal.txt": "linux_aarch64,osx_aarch64,windows_aarch64,linux_x86_64,linux_x86_64_freethreaded",
+            },
         ),
     )
     pypi = builder.build()
@@ -723,7 +725,9 @@ git_dep @ git+https://git.server/repo/project@deadbeefdeadbeef
         _parse(
             hub_name = "pypi",
             python_version = "3.15",
-            requirements_lock = "requirements.txt",
+            requirements_by_platform = {
+                "requirements.txt": "linux_aarch64,linux_x86_64,linux_x86_64_freethreaded,osx_aarch64,windows_aarch64",
+            },
             experimental_index_url = "pypi.org",
             extra_pip_args = [
                 "--extra-args-for-sdist-building",
@@ -938,18 +942,16 @@ optimum[onnxruntime-gpu]==1.17.1 ; sys_platform == 'linux'
     )
     pypi = builder.build()
 
-    # FIXME @aignas 2025-09-07: we should expose the `optimum` package
-    pypi.exposed_packages().contains_exactly([])
+    pypi.exposed_packages().contains_exactly(["optimum"])
     pypi.group_map().contains_exactly({})
     pypi.whl_map().contains_exactly({
         "optimum": {
-            "pypi_315_optimum_linux_aarch64_linux_x86_64_linux_x86_64_freethreaded": [
+            "pypi_315_optimum_linux_aarch64_linux_x86_64": [
                 whl_config_setting(
                     version = "3.15",
                     target_platforms = [
                         "cp315_linux_aarch64",
                         "cp315_linux_x86_64",
-                        "cp315_linux_x86_64_freethreaded",
                     ],
                 ),
             ],
@@ -964,7 +966,7 @@ optimum[onnxruntime-gpu]==1.17.1 ; sys_platform == 'linux'
         },
     })
     pypi.whl_libraries().contains_exactly({
-        "pypi_315_optimum_linux_aarch64_linux_x86_64_linux_x86_64_freethreaded": {
+        "pypi_315_optimum_linux_aarch64_linux_x86_64": {
             "config_load": "@pypi//:config.bzl",
             "dep_template": "@pypi//{name}:{target}",
             "python_interpreter_target": "unit_test_interpreter_target",
@@ -1019,7 +1021,9 @@ optimum[onnxruntime-gpu]==1.17.1 ; sys_platform == 'linux'
         _parse(
             hub_name = "pypi",
             python_version = "3.15",
-            requirements_lock = "universal.txt",
+            requirements_by_platform = {
+                "universal.txt": "mylinuxx86_64,myosxaarch64",
+            },
         ),
     )
     pypi = builder.build()