From 9b05b6bf5c4563686ddf25c88c62aadd5200551f Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Thu, 30 Mar 2023 23:43:18 -0500
Subject: [PATCH] fix(exe): include realesrgan and coloredlogs in bundles

---
 exe/win10.directml.dir.spec  |  3 ++-
 exe/win10.directml.file.spec | 13 ++++++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/exe/win10.directml.dir.spec b/exe/win10.directml.dir.spec
index 9b49253d..7711ebd8 100644
--- a/exe/win10.directml.dir.spec
+++ b/exe/win10.directml.dir.spec
@@ -40,6 +40,7 @@ datas = [
     ]),
     collect_data_files("onnxruntime", include_py_files=True, includes=[
         "transformers/**",
+        "tools/**",
     ]),
     collect_data_files("transformers", include_py_files=True, includes=[
         "**",
@@ -55,7 +56,7 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],
diff --git a/exe/win10.directml.file.spec b/exe/win10.directml.file.spec
index 24a7ab8e..5acd415a 100644
--- a/exe/win10.directml.file.spec
+++ b/exe/win10.directml.file.spec
@@ -31,6 +31,17 @@ datas = [
         "models/**",
         "utils/**",
     ]),
+    collect_data_files("realesrgan", include_py_files=True, includes=[
+        "archs/**",
+        "data/**",
+        "losses/**",
+        "models/**",
+        "utils/**",
+    ]),
+    collect_data_files("onnxruntime", include_py_files=True, includes=[
+        "transformers/**",
+        "tools/**",
+    ]),
     collect_data_files("transformers", include_py_files=True, includes=[
         "**",
     ]),
@@ -45,7 +56,7 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],