Skip to content

Commit b32645f

Browse files
Check in generated pyi files for some py_extension targets.
PiperOrigin-RevId: 574162025
1 parent e673985 commit b32645f

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

tensorflow_text/core/pybinds/BUILD

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,10 @@ pybind_extension(
8888
"BuildFastWordpieceModel",
8989
],
9090
copts = ["-fexceptions"],
91+
data = [
92+
"pywrap_fast_wordpiece_tokenizer_model_builder.pyi",
93+
],
94+
enable_stub_generation = True,
9195
features = ["-use_header_modules"],
9296
module_name = "pywrap_fast_wordpiece_tokenizer_model_builder",
9397
srcs_version = "PY3ONLY",
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
# ==============================================================================
15+
16+
def build_fast_wordpiece_model(arg0: list[str], arg1: int, arg2: str, arg3: str, arg4: bool, arg5: bool) -> bytes: ...

0 commit comments

Comments
 (0)