hook-sklearn.neighbors.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  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
  13. hiddenimports = []
  14. if is_module_satisfies("scikit_learn > 1.0.1"):
  15. # 1.0.2 and later
  16. hiddenimports += [
  17. 'sklearn.neighbors._quad_tree',
  18. ]
  19. elif is_module_satisfies("scikit_learn < 0.22 "):
  20. # 0.21 and below
  21. hiddenimports += [
  22. 'sklearn.neighbors.typedefs',
  23. 'sklearn.neighbors.quad_tree',
  24. ]
  25. else:
  26. # between and including 0.22 and 1.0.1
  27. hiddenimports += [
  28. 'sklearn.neighbors._typedefs',
  29. 'sklearn.neighbors._quad_tree',
  30. ]
  31. # The following hidden import must be added here
  32. # (as opposed to sklearn.tree)
  33. hiddenimports += ['sklearn.tree._criterion']
  34. # Additional hidden imports introduced in v1.0.0
  35. if is_module_satisfies("scikit_learn >= 1.0.0"):
  36. hiddenimports += ["sklearn.neighbors._partition_nodes"]