From 7e2c8707629e2f96c2451a9628b27cfee53e0e79 Mon Sep 17 00:00:00 2001
From: "David W. Berry" <dwb@greenwing.com>
Date: Thu, 13 Dec 2018 13:03:46 -0600
Subject: [PATCH] Add mlmodel to compiled file extensions (#457)

---
 Sources/XcodeGenKit/SourceGenerator.swift         | 3 ++-
 Tests/XcodeGenKitTests/SourceGeneratorTests.swift | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Sources/XcodeGenKit/SourceGenerator.swift b/Sources/XcodeGenKit/SourceGenerator.swift
index 129e1653..b3b86a20 100644
--- a/Sources/XcodeGenKit/SourceGenerator.swift
+++ b/Sources/XcodeGenKit/SourceGenerator.swift
@@ -192,7 +192,8 @@ class SourceGenerator {
                  "S",
                  "xcdatamodeld",
                  "intentdefinition",
-                 "metal":
+                 "metal",
+                 "mlmodel":
                 return .sources
             case "h",
                  "hh",
diff --git a/Tests/XcodeGenKitTests/SourceGeneratorTests.swift b/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
index 5054dde8..e68c3311 100644
--- a/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
+++ b/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
@@ -435,6 +435,7 @@ class SourceGeneratorTests: XCTestCase {
                     - file.123
                     - file.xcassets
                     - file.metal
+                    - file.mlmodel
                     - Info.plist
                     - Intent.intentdefinition
                 """
@@ -484,6 +485,7 @@ class SourceGeneratorTests: XCTestCase {
                 try pbxProj.expectFile(paths: ["C", "file.123"], buildPhase: .resources)
                 try pbxProj.expectFile(paths: ["C", "Info.plist"], buildPhase: .none)
                 try pbxProj.expectFile(paths: ["C", "file.metal"], buildPhase: .sources)
+                try pbxProj.expectFile(paths: ["C", "file.mlmodel"], buildPhase: .sources)
                 try pbxProj.expectFile(paths: ["C", "Intent.intentdefinition"], buildPhase: .sources)
             }