Fix missing origin for mlmodelc files (#2799)

Bundling a .mlmodel file into a resource bundle results in the following
error:

```
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/testing/ios_rules.bzl", line 77, column 60, in _ios_unit_test_bundle_impl
		return apple_test_bundle_support.apple_test_bundle_impl(
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/testing/apple_test_bundle_support.bzl", line 545, column 41, in _apple_test_bundle_impl
		processor_result = processor.process(
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/processor.bzl", line 748, column 36, in _process
		partial_outputs = [partial.call(p) for p in partials]
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/bazel_skylib+/lib/partial.bzl", line 43, column 28, in _call
		return partial.function(*function_args, **function_kwargs)
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/partials/resources.bzl", line 299, column 26, in _resources_partial_impl
		resources.deduplicate(
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/resources.bzl", line 997, column 42, in _deduplicate
		deduplicated = _deduplicate_field(
	File "/private/var/tmp/_bazel_johnflanagan/a29065da0d532a50315a8974596a3ed4/external/rules_apple+/apple/internal/resources.bzl", line 942, column 49, in _deduplicate_field
		path_origins = processed_origins[short_path]
Error: key "example.resource_bundle-intermediates/Example_Resources.bundle/FastViTT8F16.mlmodelc" not found in dictionary
```

I made an example repo to demonstrate the issue here:
https://github.com/jflan-dd/coreml-FastViT-T8

This fix adds the `.mlmodel` to `.mlmodelc` mapping to
`processed_origins`

---------

Co-authored-by: Aaron Sky <aaronsky@skyaaron.com>

Author: jflan-dd

apple/internal/partials/support/resources_support.bzl

@@ -471,6 +471,8 @@ def _mlmodels(
             output_discriminator = output_discriminator,
             dir_name = paths.join(parent_dir or "", paths.replace_extension(basename, ".mlmodelc")),
         )
+        processed_origins[output_bundle.short_path] = [file.short_path]
+
         output_plist = intermediates.file(
             actions = actions,
             target_name = rule_label.name,

test/BUILD

@@ -73,6 +73,15 @@ apple_shell_test(
     ],
 )
 
+apple_shell_test(
+    name = "apple_core_ml_resource_test",
+    size = "medium",
+    src = "apple_core_ml_resource_test.sh",
+    data = [
+        "//test/testdata/resources:mlmodel_srcs",
+    ],
+)
+
 apple_shell_test(
     name = "apple_intent_library_test",
     size = "medium",

test/apple_core_ml_resource_test.sh

@@ -0,0 +1,92 @@
+#!/bin/bash
+
+# Copyright 2019 The Bazel Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -eu
+
+# Integration tests for bundling CoreML models into resource bundles.
+
+function set_up() {
+  mkdir -p app
+}
+
+function tear_down() {
+  rm -rf app
+}
+
+# Creates common source, targets, and basic plist for iOS applications.
+function create_common_files() {
+  cat > app/BUILD <<EOF
+load("@build_bazel_rules_apple//apple:ios.bzl", "ios_application")
+load(
+    "@build_bazel_rules_apple//apple:resources.bzl",
+    "apple_precompiled_resource_bundle",
+  )
+load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
+
+apple_precompiled_resource_bundle(
+    name = "app.resources",
+    bundle_name = "App_Resources",
+    resources = [
+        "@build_bazel_rules_apple//test/testdata/resources:sample.mlpackage",
+    ],
+)
+
+swift_library(
+    name = "app_lib",
+    data = ["app.resources"],
+    srcs = ["AppDelegate.swift"],
+)
+
+ios_application(
+    name = "app",
+    bundle_id = "my.bundle.id",
+    families = ["iphone", "ipad"],
+    infoplists = ["Info.plist"],
+    minimum_os_version = "${MIN_OS_IOS}",
+    deps = [":app_lib"],
+)
+EOF
+
+  cat > app/AppDelegate.swift <<EOF
+import UIKit
+
+@UIApplicationMain
+class AppDelegate: UIResponder, UIApplicationDelegate {
+  var window: UIWindow?
+}
+EOF
+
+  cat > app/Info.plist <<EOF
+{
+  CFBundleIdentifier = "\${PRODUCT_BUNDLE_IDENTIFIER}";
+  CFBundleName = "\${PRODUCT_NAME}";
+  CFBundlePackageType = "APPL";
+  CFBundleShortVersionString = "1.0";
+  CFBundleVersion = "1.0";
+}
+EOF
+}
+
+# Tests that the mlpackage is compiled correctly into a resource bundle
+function test_mlmodel_resource_bundle_builds() {
+  create_common_files
+
+  do_build ios //app:app || fail "Should build"
+
+  assert_zip_contains "test-bin/app/app.ipa" "Payload/app.app/App_Resources.bundle/sample.mlmodelc/"
+}
+
+run_suite "coreml resource bundle tests"