hook-sklearn.utils.py 749 B

1234567891011121314151617181920
  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.utils._cython_blas']
  14. # As of scikit-learn 1.7.1, the `sklearn.utils._isfinite` extension started to depend on newly-introduced
  15. # `sklearn._cyutility`.
  16. if is_module_satisfies('scikit-learn >= 1.7.1'):
  17. hiddenimports += ['sklearn._cyutility']