hook-skimage.feature.py 1.3 KB

1234567891011121314151617181920212223242526
  1. # ------------------------------------------------------------------
  2. # Copyright (c) 2020 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 is_module_satisfies, collect_data_files, collect_submodules
  13. # The following missing module prevents import of skimage.feature with skimage 0.17.x.
  14. hiddenimports = ['skimage.feature._orb_descriptor_positions', ]
  15. # Collect the data file with ORB descriptor positions. In earlier versions of scikit-image, this file was in
  16. # `skimage/data` directory, and it was moved to `skimage/feature` in v0.17.0. Collect if from wherever it is.
  17. datas = collect_data_files('skimage', includes=['**/orb_descriptor_positions.txt'])
  18. # As of scikit-image 0.22.0, we need to collect the __init__.pyi file for `lazy_loader`, as well as collect submodules
  19. # due to lazy loading.
  20. if is_module_satisfies("scikit-image >= 0.22.0"):
  21. datas += collect_data_files("skimage.feature", includes=["*.pyi"])
  22. hiddenimports = collect_submodules('skimage.feature', filter=lambda name: name != 'skimage.feature.tests')