-- MODULE.bazel --
module(name = "py_imports")

rules_python_version = "0.24.0"

bazel_dep(name = "rules_python", version = rules_python_version)
bazel_dep(name = "rules_python_gazelle_plugin", version = rules_python_version)
bazel_dep(name = "gazelle", version = "0.31.0", repo_name = "bazel_gazelle")

python = use_extension("@rules_python//python/extensions:python.bzl", "python")
python.toolchain(
    configure_coverage_tool = True,
    is_default = True,
    python_version = "3.9",
)
-- WORKSPACE --
-- pkg1/BUILD.bazel --
load("@rules_python//python:defs.bzl", "py_binary")

py_binary(
    name = "pkg1",
    srcs = ["__init__.py", "main.py"],
    main = "main.py",
    deps = ["//pkg2"],
)
-- pkg1/__init__.py --
-- pkg1/main.py --
from pkg2 import lib

print(lib.hello())
-- pkg2/BUILD.bazel --
load("@rules_python//python:defs.bzl", "py_library")

py_library(
    name = "pkg2",
    srcs = ["__init__.py", "lib.py"],
    visibility = ["//visibility:public"],
)
-- pkg2/__init__.py --
-- pkg2/lib.py --
def hello():
    return "hello world"

bazel run --enable_bzlmod //pkg1
INFO: Analyzed target //pkg1:pkg1 (0 packages loaded, 0 targets configured).
INFO: Found 1 target...
Target //pkg1:pkg1 up-to-date:
  bazel-bin/pkg1/pkg1
INFO: Elapsed time: 0.125s, Critical Path: 0.01s
INFO: 1 process: 1 internal.
INFO: Build completed successfully, 1 total action
INFO: Running command line: bazel-bin/pkg1/pkg1
hello world