Add site-package example

Author: hartikainen

examples/site-packages/.bazelrc

@@ -0,0 +1,6 @@
+common --@rules_python//python/config_settings:bootstrap_impl=script
+common --@rules_python//python/config_settings:venvs_site_packages=yes
+# See https://github.com/bazel-contrib/rules_python/issues/2864#issuecomment-2859325467
+# common --@rules_python//python/config_settings:venvs_use_declare_symlink=no
+
+build --incompatible_default_to_explicit_init_py=true

examples/site-packages/.bazelversion

@@ -0,0 +1 @@
+8.2.1

examples/site-packages/BUILD.bazel

@@ -0,0 +1,29 @@
+load("@rules_python//python:defs.bzl", "py_test")
+
+
+py_test(
+    name = "test",
+    srcs = ["test.py"],
+    deps = [
+        "@pypi//mujoco",
+        # "@pypi//mujoco_mjx",
+        # "@pypi//tensorflow",
+        # "@pypi//importlib_resources",
+        # "@pypi//typing_extensions",  # ModuleNotFoundError: No module named 'typing_extensions'
+        "@pypi//jax",
+        "@pypi//jax_cuda12_pjrt",
+        "@pypi//jax_cuda12_plugin",
+        "@pypi//nvidia_cublas_cu12",
+        "@pypi//nvidia_cuda_cupti_cu12",
+        "@pypi//nvidia_cuda_nvcc_cu12",
+        "@pypi//nvidia_cuda_nvrtc_cu12",
+        "@pypi//nvidia_cuda_runtime_cu12",
+        "@pypi//nvidia_cudnn_cu12",
+        "@pypi//nvidia_cufft_cu12",
+        "@pypi//nvidia_cusolver_cu12",
+        "@pypi//nvidia_cusparse_cu12",
+        "@pypi//nvidia_nccl_cu12",
+        "@pypi//nvidia_nvjitlink_cu12",
+        "@pypi//nvidia_nvshmem_cu12",
+    ],
+)

examples/site-packages/MODULE.bazel

@@ -0,0 +1,32 @@
+"""Site-packages example."""
+
+module(
+    name = "site-packages-example",
+    version = "0.0.1",
+)
+
+bazel_dep(name = "rules_python", version = "0.0.0")
+bazel_dep(name = "platforms", version = "0.0.11")
+
+local_path_override(
+    module_name = "rules_python",
+    path = "../..",
+)
+
+python = use_extension("@rules_python//python/extensions:python.bzl", "python")
+python.toolchain(
+    is_default = True,
+    python_version = "3.12",
+)
+python.override(minor_mapping = {"3.12": "3.12.7"})
+
+pip = use_extension("@rules_python//python/extensions:pip.bzl", "pip")
+pip.parse(
+    experimental_index_url = "https://pypi.org/simple",
+    hub_name = "pypi",
+    python_version = "3.12",
+    requirements_lock = "requirements.txt",
+    # enable_implicit_namespace_pkgs = False,
+)
+
+use_repo(pip, "pypi")

examples/site-packages/requirements.in

@@ -0,0 +1,5 @@
+mujoco
+# mujoco-mjx
+jax[cuda12]==0.7.1
+jaxlib==0.7.1
+tensorflow

examples/site-packages/requirements.txt

@@ -0,0 +1,20 @@
+"""A simple site-packages test."""
+
+import jax
+import mujoco
+# `mjx` does not work because "it's a prefix of the other."
+# from mujoco import mjx
+import nvidia
+
+
+def main() -> None:
+    print(f"Hello, {nvidia=}!")
+    print(f"Hello, {jax=}!")
+    print(f"{mujoco=}!")
+    # print(f"{mjx=}")
+
+    print(f"{jax.devices('gpu')=}")
+
+
+if __name__ == "__main__":
+    main()