Skip to content

GitLab

Unverified Commit a6bf4ad4 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[python-package] fix mypy error about cpu_count() methods (#5786)

parent 54486b4f
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
python-package/lightgbm/compat.py
View file @ a6bf4ad4
...@@ -183,18 +183,18 @@ except ImportError: ...@@ -183,18 +183,18 @@ except ImportError:
try: try:
from joblib import cpu_count from joblib import cpu_count
def _LGBMCpuCount(only_physical_cores: bool = True): def _LGBMCpuCount(only_physical_cores: bool = True) -> int:
return cpu_count(only_physical_cores=only_physical_cores) return cpu_count(only_physical_cores=only_physical_cores)
except ImportError: except ImportError:
try: try:
from psutil import cpu_count from psutil import cpu_count
def _LGBMCpuCount(only_physical_cores: bool = True): def _LGBMCpuCount(only_physical_cores: bool = True) -> int:
return cpu_count(logical=not only_physical_cores) return cpu_count(logical=not only_physical_cores) or 1
except ImportError: except ImportError:
from multiprocessing import cpu_count from multiprocessing import cpu_count
def _LGBMCpuCount(only_physical_cores: bool = True): def _LGBMCpuCount(only_physical_cores: bool = True) -> int:
return cpu_count() return cpu_count()
__all__: List[str] = [] __all__: List[str] = []
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment