1load("@fbcode_macros//build_defs:python_library.bzl", "python_library") 2load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 3 4oncall("ai_infra_mobile_platform") 5 6python_library( 7 name = "lib", 8 srcs = [ 9 "api.py", 10 "sample_input.py", 11 ], 12 deps = [ 13 "//caffe2:torch", 14 "//caffe2/torchgen:torchgen", 15 "//executorch/exir/dialects/edge/arg:lib", 16 "//executorch/exir/operator:convert", 17 ], 18) 19 20python_unittest( 21 name = "test_api", 22 srcs = [ 23 "test/test_api.py", 24 ], 25 deps = [ 26 ":lib", 27 "//caffe2:torch", 28 "//caffe2/torchgen:torchgen", 29 ], 30) 31