Skip to content

Commit d5d8680

Browse files
authored
Only run python test in fbcode
Differential Revision: D78122200 Pull Request resolved: #100
1 parent 4c05a45 commit d5d8680

File tree

2 files changed

+23
-20
lines changed

2 files changed

+23
-20
lines changed

test/TARGETS

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,15 @@ oncall("ai_infra_mobile_platform")
88

99
define_common_targets()
1010

11+
runtime.export_file(
12+
name = "test_tiktoken_tokenizer_model",
13+
src = "resources/test_tiktoken_tokenizer.model",
14+
visibility = [
15+
"//pytorch/tokenizers/...",
16+
"@EXECUTORCH_CLIENTS",
17+
],
18+
)
19+
1120
runtime.export_file(
1221
name = "test_hf_tokenizer.json",
1322
src = "resources/test_hf_tokenizer.json",
@@ -17,6 +26,20 @@ runtime.export_file(
1726
],
1827
)
1928

29+
runtime.python_test(
30+
name = "test_tiktoken_py",
31+
srcs = [
32+
"test_tiktoken.py",
33+
],
34+
resources = {
35+
":test_tiktoken_tokenizer_model": "resources/test_tiktoken_tokenizer.model",
36+
},
37+
deps = [
38+
"fbsource//third-party/pypi/blobfile:blobfile",
39+
"//pytorch/tokenizers/pytorch_tokenizers:tokenizers",
40+
],
41+
)
42+
2043
runtime.python_test(
2144
name = "test_python_bindings",
2245
srcs = [

test/targets.bzl

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -114,26 +114,6 @@ def define_common_targets():
114114
]),
115115
)
116116

117-
runtime.export_file(
118-
name = "test_tiktoken_tokenizer_model",
119-
src = "resources/test_tiktoken_tokenizer.model",
120-
visibility = ["@EXECUTORCH_CLIENTS", "//pytorch/tokenizers/..."],
121-
)
122-
123-
runtime.python_test(
124-
name = "test_tiktoken_py",
125-
srcs = [
126-
"test_tiktoken.py",
127-
],
128-
deps = [
129-
"//pytorch/tokenizers/pytorch_tokenizers:tokenizers",
130-
"fbsource//third-party/pypi/blobfile:blobfile",
131-
],
132-
resources = {
133-
":test_tiktoken_tokenizer_model": "resources/test_tiktoken_tokenizer.model",
134-
},
135-
)
136-
137117
runtime.cxx_test(
138118
name = "test_hf_tokenizer",
139119
srcs = [

0 commit comments

Comments
 (0)