Add type stubs

PiperOrigin-RevId: 896018428

Author: CEL Dev Team

cel_expr_python/BUILD

@@ -1,6 +1,6 @@
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension", "pybind_library")
 load("@rules_cc//cc:cc_library.bzl", "cc_library")
-load("@rules_python//python:defs.bzl", "py_test")
+load("@rules_python//python:py_test.bzl", "py_test")
 
 package(default_visibility = ["//visibility:private"])
 
@@ -45,6 +45,7 @@ pybind_extension(
         "py_message_factory.cc",
         "py_message_factory.h",
     ],
+    data = ["cel.pyi"],
     visibility = ["//visibility:public"],
     deps = [
         ":cel_extension",

cel_expr_python/ext/BUILD

@@ -8,6 +8,7 @@ pybind_extension(
         "ext_bindings.cc",
     ],
     data = [
+        "ext_bindings.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],
@@ -26,6 +27,7 @@ pybind_extension(
         "ext_encoders.cc",
     ],
     data = [
+        "ext_encoders.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],
@@ -47,6 +49,7 @@ pybind_extension(
         "ext_math.cc",
     ],
     data = [
+        "ext_math.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],
@@ -80,6 +83,7 @@ pybind_extension(
         "ext_optional.cc",
     ],
     data = [
+        "ext_optional.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],
@@ -101,6 +105,7 @@ pybind_extension(
         "ext_proto.cc",
     ],
     data = [
+        "ext_proto.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],
@@ -119,6 +124,7 @@ pybind_extension(
         "ext_string.cc",
     ],
     data = [
+        "ext_string.pyi",
         "//cel_expr_python:cel",
     ],
     visibility = ["//visibility:public"],

cel_expr_python/ext/ext_bindings.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtBindings(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

cel_expr_python/ext/ext_encoders.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtEncoders(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

cel_expr_python/ext/ext_math.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtMath(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

cel_expr_python/ext/ext_optional.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtOptional(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

cel_expr_python/ext/ext_proto.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtProto(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

cel_expr_python/ext/ext_string.pyi

@@ -0,0 +1,4 @@
+from .. import cel
+
+class ExtString(cel.CelExtensionBase):
+  def __init__(self) -> None: ...

conformance/BUILD

@@ -1,4 +1,4 @@
-load("@rules_python//python:defs.bzl", "py_test")
+load("@rules_python//python:py_test.bzl", "py_test")
 
 package(default_visibility = ["//visibility:private"])
 

custom_ext/BUILD

@@ -1,13 +1,14 @@
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
-load("@rules_python//python:defs.bzl", "py_test")
 load("@rules_python//python:py_library.bzl", "py_library")
+load("@rules_python//python:py_test.bzl", "py_test")
 
 package(default_visibility = ["//visibility:private"])
 
 pybind_extension(
     name = "sample_cel_ext_cc",
     srcs = ["sample_cel_ext.cc"],
     data = [
+        "sample_cel_ext.pyi",
         "//cel_expr_python:cel",
     ],
     deps = [