pbar have unique id
This commit is contained in:
@@ -24,7 +24,12 @@ def fingerprint(params: BareParams):
|
||||
|
||||
|
||||
def plot_all(sim_dir: Path, limits: list[str], **opts):
|
||||
for k, v in opts.items():
|
||||
if k in ["skip"]:
|
||||
opts[k] = int(v)
|
||||
dir_list = list(p for p in sim_dir.glob("*") if p.is_dir())
|
||||
if len(dir_list) == 0:
|
||||
dir_list = [sim_dir]
|
||||
limits = [
|
||||
tuple(func(el) for func, el in zip([float, float, str], lim.split(","))) for lim in limits
|
||||
]
|
||||
|
||||
@@ -17,6 +17,7 @@ from pathlib import Path
|
||||
from typing import Any, Dict, Iterable, Iterator, List, Tuple, TypeVar, Union
|
||||
|
||||
import numpy as np
|
||||
import random
|
||||
from tqdm import tqdm
|
||||
|
||||
|
||||
@@ -38,6 +39,8 @@ class PBars:
|
||||
worker_kwargs=None,
|
||||
) -> "PBars":
|
||||
|
||||
self.id = random.randint(100000, 999999)
|
||||
|
||||
if isinstance(task, abc.Iterable):
|
||||
self.iterator: Iterable[T_] = iter(task)
|
||||
self.num_tot: int = len(task)
|
||||
@@ -63,7 +66,7 @@ class PBars:
|
||||
if "desc" in kwargs:
|
||||
kwargs["desc"] = kwargs["desc"].format(worker_id=i)
|
||||
self.append(tqdm(position=i, ncols=100, ascii=False, **kwargs))
|
||||
self.print_path = Path("progress " + self.pbars[0].desc).resolve()
|
||||
self.print_path = Path(f"progress {self.pbars[0].desc} {self.id}").resolve()
|
||||
self.close_ev = threading.Event()
|
||||
if "file" in self.policy:
|
||||
self.thread = threading.Thread(target=self.print_worker, daemon=True)
|
||||
|
||||
Reference in New Issue
Block a user