agi-med-common 3.5.13__tar.gz → 3.6.1__tar.gz

{agi_med_common-3.5.13 → agi_med_common-3.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 3.5.13
+Version: 3.6.1
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-3.5.13 → agi_med_common-3.6.1}/src/agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.5.13"
+__version__ = "3.6.1"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -18,3 +18,4 @@ from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnu
 from .utils import make_session_id, read_json
 from .validators import is_file_exist, validate_prompt
 from .xml_parser import XMLParser
+from .parallel_map import parallel_map

agi_med_common-3.6.1/src/agi_med_common/parallel_map.py

@@ -0,0 +1,31 @@
+from typing import Any, Callable, TypeVar
+import concurrent.futures
+from tqdm import tqdm
+
+
+X = TypeVar("X")
+
+
+def parallel_map(
+    func: Callable[..., X],
+    items: list[Any],
+    process: bool = False,
+    multiple_args: bool = False,
+    max_workers: int = 2,
+    show_tqdm: bool = False,
+    desc: str = "",
+) -> list[X]:
+    pool = (
+        concurrent.futures.ProcessPoolExecutor(max_workers=max_workers)
+        if process
+        else concurrent.futures.ThreadPoolExecutor(max_workers=max_workers)
+    )
+    with pool as executor:
+        futures = []
+        for item in items:
+            if multiple_args:
+                futures.append(executor.submit(func, *item))
+            else:
+                futures.append(executor.submit(func, item))
+        results: list[X] = [future.result() for future in tqdm(futures, disable=(not show_tqdm), desc=desc)]
+    return results

{agi_med_common-3.5.13 → agi_med_common-3.6.1}/src/agi_med_common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 3.5.13
+Version: 3.6.1
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-3.5.13 → agi_med_common-3.6.1}/src/agi_med_common.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ pyproject.toml
 requirements.txt
 src/agi_med_common/__init__.py
 src/agi_med_common/file_storage.py
+src/agi_med_common/parallel_map.py
 src/agi_med_common/utils.py
 src/agi_med_common/validators.py
 src/agi_med_common/xml_parser.py