diff --git a/mediapipe/tasks/ios/ios.bzl b/mediapipe/tasks/ios/ios.bzl new file mode 100644 index 000000000..ad0f865ad --- /dev/null +++ b/mediapipe/tasks/ios/ios.bzl @@ -0,0 +1,15 @@ +"""Mediapipe Task Library Helper Rules for iOS""" + +MPP_TASK_MINIMUM_OS_VERSION = "11.0" + +# Default tags for filtering iOS targets. Targets are restricted to Apple platforms. +MPP_TASK_DEFAULT_TAGS = [ + "apple", +] + +# Following sanitizer tests are not supported by iOS test targets. +MPP_TASK_DISABLED_SANITIZER_TAGS = [ + "noasan", + "nomsan", + "notsan", +] diff --git a/mediapipe/tasks/ios/test/text/text_classifier/BUILD b/mediapipe/tasks/ios/test/text/text_classifier/BUILD index 81479e68b..2202ff1a6 100644 --- a/mediapipe/tasks/ios/test/text/text_classifier/BUILD +++ b/mediapipe/tasks/ios/test/text/text_classifier/BUILD @@ -1,31 +1,22 @@ -load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test") -load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner") +load( + "//mediapipe/tasks:ios/ios.bzl", + "MPP_TASK_MINIMUM_OS_VERSION", + "MPP_TASK_DEFAULT_TAGS", + "MPP_TASK_DISABLED_SANITIZER_TAGS", +) +load( + "@build_bazel_rules_apple//apple:ios.bzl", + "ios_unit_test", +) +load( + "@org_tensorflow//tensorflow/lite:special_rules.bzl", + "tflite_ios_lab_runner" +) package(default_visibility = ["//mediapipe/tasks:internal"]) licenses(["notice"]) -TFL_MINIMUM_OS_VERSION = "11.0" -# LINT.ThenChange( -# TensorFlowLiteC.podspec.template, -# TensorFlowLiteSelectTfOps.podspec.template, -# ../objc/TensorFlowLiteObjC.podspec.template, -# ../swift/TensorFlowLiteSwift.podspec.template -# ) - -# Default tags for filtering iOS targets. Targets are restricted to Apple platforms. -TFL_DEFAULT_TAGS = [ - "apple", -] - -# Following sanitizer tests are not supported by iOS test targets. -TFL_DISABLED_SANITIZER_TAGS = [ - "noasan", - "nomsan", - "notsan", -] - - objc_library( name = "MPPTextClassifierObjcTestLibrary", testonly = 1, @@ -34,7 +25,7 @@ objc_library( "//mediapipe/tasks/testdata/text:bert_text_classifier_models", "//mediapipe/tasks/testdata/text:text_classifier_models", ], - tags = TFL_DEFAULT_TAGS, + tags = MPP_TASK_DEFAULT_TAGS, copts = [ "-ObjC++", "-std=c++17", @@ -47,9 +38,9 @@ objc_library( ios_unit_test( name = "MPPTextClassifierObjcTest", - minimum_os_version = TFL_MINIMUM_OS_VERSION, + minimum_os_version = MPP_TASK_MINIMUM_OS_VERSION, runner = tflite_ios_lab_runner("IOS_LATEST"), - tags = TFL_DEFAULT_TAGS + TFL_DISABLED_SANITIZER_TAGS, + tags = MPP_TASK_DEFAULT_TAGS + MPP_TASK_DISABLED_SANITIZER_TAGS, deps = [ ":MPPTextClassifierObjcTestLibrary", ],