Requires PyInstaller 6.x

news/666.update.rst

@@ -1 +1 @@
-Discard header files (and static libraries on Windows) of ``torch``
+Modernize the torch for ``hook`` and reduce the amount of unnecessarily collected data files (header files and static libraries). Requires PyInstaller >= 6.0.

src/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-torch.py

@@ -10,19 +10,40 @@
 # SPDX-License-Identifier: GPL-2.0-or-later
 # ------------------------------------------------------------------
 
-from PyInstaller.utils.hooks import logger, collect_data_files, is_module_satisfies, collect_dynamic_libs, collect_submodules
+from PyInstaller.utils.hooks import (
+    logger,
+    collect_data_files,
+    is_module_satisfies,
+    collect_dynamic_libs,
+    collect_submodules,
+    get_package_paths,
+)
 
-module_collection_mode = 'pyz+py'
+if is_module_satisfies("PyInstaller >= 6.0"):
+    module_collection_mode = "pyz+py"
 
-datas = collect_data_files("torch", excludes=["**/*.h", "**/*.hpp", "**/*.cuh",
-                                              "**/*.lib", "**/*.cpp", "**/*.pyi", "**/*.cmake"])
-binaries = collect_dynamic_libs("torch")
-hiddenimports = collect_submodules("torch")
+    datas = collect_data_files(
+        "torch",
+        excludes=[
+            "**/*.h",
+            "**/*.hpp",
+            "**/*.cuh",
+            "**/*.lib",
+            "**/*.cpp",
+            "**/*.pyi",
+            "**/*.cmake",
+        ],
+    )
+    binaries = collect_dynamic_libs("torch")
+    hiddenimports = collect_submodules("torch")
+else:
+    datas = [(get_package_paths("torch")[1], "torch")]
 
 # With torch 2.0.0, PyInstaller's modulegraph analysis hits the recursion limit.
 # So, unless the user has already done so, increase it automatically.
-if is_module_satisfies('torch >= 2.0.0'):
+if is_module_satisfies("torch >= 2.0.0"):
     import sys
+
     new_limit = 5000
     if sys.getrecursionlimit() < new_limit:
         logger.info("hook-torch: raising recursion limit to %d", new_limit)