Add pytype_srcs to main jax BUILD rule

PiperOrigin-RevId: 476989241
google · Sep 26, 2022 · 265b39d · 265b39d
1 parent ddd8581
commit 265b39d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jax/BUILD b/jax/BUILD
@@ -110,6 +110,7 @@ py_library_providing_imports_info(
         "experimental/compilation_cache/cache_interface.py",
     ],
     lib_rule = pytype_library,
+    pytype_srcs = glob(["_src/**/*.pyi"]),
     visibility = ["//visibility:public"],
     deps = select({
                ":enable_jaxlib_build": [":jaxlib_deps"],