From 25e0ece8e2cbb76d488ed7cbe8b11e53418a9c15 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 30 Apr 2023 17:19:29 -0500
Subject: [PATCH] fix(exe): add pytorch_lightning to Windows bundle (#366)

---
 exe/build.bat                | 5 +++--
 exe/win10.directml.dir.spec  | 8 +++++++-
 exe/win10.directml.file.spec | 8 +++++++-
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/exe/build.bat b/exe/build.bat
index 85573ffd..97bc5f75 100644
--- a/exe/build.bat
+++ b/exe/build.bat
@@ -16,7 +16,8 @@ xcopy ..\models .\dist\models /s /e /f /i /y
 xcopy ..\outputs .\dist\outputs /s /e /f /i /y
 
 REM copy loose files
-copy ".\onnx-web-*" .\dist\ \y
+copy .\onnx-web-full.bat .\dist\onnx-web-full.bat \y
+copy .\onnx-web-half.bat .\dist\onnx-web-half.bat \y
 copy .\README.txt .\dist\README.txt
 copy ..\api\logging.yaml .\dist\logging.yaml /y
 copy ..\api\params.json .\dist\params.json /y
@@ -31,4 +32,4 @@ set /p GIT_SHA=<commit.txt
 set GIT_HEAD=%GIT_SHA:~0,8%
 
 REM create archive
-"C:\Program Files\7-Zip\7z.exe" a ..\dist\onnx-web-v%BUNDLE_VERSION%-%BUNDLE_TYPE%-%GIT_HEAD%.zip ".\dist\*"
+"C:\Program Files\7-Zip\7z.exe" a .\dist\onnx-web-v%BUNDLE_VERSION%-%BUNDLE_TYPE%-%GIT_HEAD%.zip ".\dist\*"
diff --git a/exe/win10.directml.dir.spec b/exe/win10.directml.dir.spec
index 7711ebd8..3a771974 100644
--- a/exe/win10.directml.dir.spec
+++ b/exe/win10.directml.dir.spec
@@ -56,7 +56,13 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=[
+        'coloredlogs',
+        'onnxruntime',
+        'onnxruntime-directml',
+        'pytorch_lightning',
+        'tqdm'
+    ],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],
diff --git a/exe/win10.directml.file.spec b/exe/win10.directml.file.spec
index 5acd415a..08d190f6 100644
--- a/exe/win10.directml.file.spec
+++ b/exe/win10.directml.file.spec
@@ -56,7 +56,13 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=[
+        'coloredlogs',
+        'onnxruntime',
+        'onnxruntime-directml',
+        'pytorch_lightning',
+        'tqdm'
+    ],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],