hook-sklearn.tree.py 619 B

123456789101112131415161718
  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 = ['sklearn.tree._utils']
  14. if is_module_satisfies('scikit-learn >= 1.6.0'):
  15. hiddenimports += ['sklearn.tree._partitioner']