hook-torchaudio.py 867 B

123456789101112131415161718192021
  1. # ------------------------------------------------------------------
  2. # Copyright (c) 2023 PyInstaller Development Team.
  3. #
  4. # This file is distributed under the terms of the GNU General Public
  5. # License (version 2.0 or later).
  6. #
  7. # The full license is available in LICENSE, distributed with
  8. # this software.
  9. #
  10. # SPDX-License-Identifier: GPL-2.0-or-later
  11. # ------------------------------------------------------------------
  12. from PyInstaller.utils.hooks import collect_dynamic_libs, collect_submodules
  13. # Collect dynamic extensions from torchaudio/lib - some of them are loaded dynamically, and are thus not automatically
  14. # collected.
  15. binaries = collect_dynamic_libs('torchaudio')
  16. hiddenimports = collect_submodules('torchaudio.lib')
  17. # Collect source .py files for JIT/torchscript. Requires PyInstaller >= 5.3, no-op in older versions.
  18. module_collection_mode = 'pyz+py'