andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Remove psutil

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-25 19:04:13 +02:00
parent 38616b1c4f
commit 394d4465b0
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
2 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
.vscode/settings.json vendored
View file

@ -31,7 +31,6 @@
"Parcoords", "Parcoords",
"plotly", "plotly",
"proba", "proba",
"psutil",
"pydantic", "pydantic",
"pymongo", "pymongo",
"pyplot", "pyplot",

3
src/great_ai/utilities/parallel_map.py
View file

@ -2,7 +2,6 @@ from math import ceil
from typing import Any, Callable, Iterable, List, Optional from typing import Any, Callable, Iterable, List, Optional
import multiprocess as mp import multiprocess as mp
import psutil
from tqdm.cli import tqdm from tqdm.cli import tqdm
from .logger import get_logger from .logger import get_logger
@ -18,7 +17,7 @@ def parallel_map(
disable_progress: bool = False, disable_progress: bool = False,
) -> List[Any]: ) -> List[Any]:
if concurrency is None: if concurrency is None:
concurrency = psutil.cpu_count() concurrency = mp.cpu_count()
assert concurrency > 0 assert concurrency > 0
assert chunk_size is None or chunk_size > 0 assert chunk_size is None or chunk_size > 0