hydraflow 0.2.10__py3-none-any.whl → 0.2.11__py3-none-any.whl

hydraflow/progress.py

@@ -161,21 +161,27 @@ def multi_tasks_progress(
 
     with Progress(*columns, transient=transient or False, **kwargs) as progress:
         task_main = progress.add_task(main_description, total=None)
+
+        task_ids = [
+            progress.add_task(description.format(i), start=False, total=None)
+            for i in range(len(iterables))
+        ]
+
         total = {}
         completed = {}
 
-        def func(i: int, iterable: Iterable[int | tuple[int, int]]) -> None:
-            task_id = progress.add_task(description.format(i), total=None)
+        def func(i: int) -> None:
             completed[i] = 0
             total[i] = None
+            progress.start_task(task_ids[i])
 
-            for index in iterable:
+            for index in iterables[i]:
                 if isinstance(index, tuple):
                     completed[i], total[i] = index[0] + 1, index[1]
                 else:
                     completed[i] = index + 1
 
-                progress.update(task_id, total=total[i], completed=completed[i])
+                progress.update(task_ids[i], total=total[i], completed=completed[i])
 
                 if all(t is not None for t in total.values()):
                     t = sum(total.values())
@@ -185,7 +191,7 @@ def multi_tasks_progress(
                 progress.update(task_main, total=t, completed=c)
 
             if transient is not False:
-                progress.remove_task(task_id)
+                progress.remove_task(task_ids[i])
 
-        it = (joblib.delayed(func)(i, it) for i, it in enumerate(iterables))
+        it = (joblib.delayed(func)(i) for i in range(len(iterables)))
         joblib.Parallel(n_jobs, prefer="threads")(it)

{hydraflow-0.2.10.dist-info → hydraflow-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.2.10
+Version: 0.2.11
 Summary: Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments.
 Project-URL: Documentation, https://github.com/daizutabi/hydraflow
 Project-URL: Source, https://github.com/daizutabi/hydraflow

{hydraflow-0.2.10.dist-info → hydraflow-0.2.11.dist-info}/RECORD

@@ -4,9 +4,9 @@ hydraflow/config.py,sha256=6TCKNQZ3sSrIEvl245T2udwFuknejyN1dMcIVmOHdrQ,2102
 hydraflow/context.py,sha256=G7JMrG70sgBH2qILXl5nkGWNUoRggj518JWUq0ZiJ9E,7776
 hydraflow/info.py,sha256=Vj2sT66Ric63mmaq7Yu8nDFhsGQYO3MCHrxFpapDufc,3458
 hydraflow/mlflow.py,sha256=Q8RGijSURTjRkEDxzi_2Tk9KOx3QK__al5aArGQriHA,7249
-hydraflow/progress.py,sha256=hinjOnz29DJj7HskW21XyrkheaWtScj7th0MYFCQIbM,6537
+hydraflow/progress.py,sha256=uaxS9UnwsikOPB_Ho-5mXdBYz3Zedj-SL9FzT094wjY,6623
 hydraflow/run_collection.py,sha256=V5lGdGHYgsSpBOYGaVEL1mpKJvdiEshBL0KmmZ8qeZo,29161
-hydraflow-0.2.10.dist-info/METADATA,sha256=dyQTcKSv2A0sKgEqSWXpkFYN9HA-0tFhRF5qXjaVMS0,4182
-hydraflow-0.2.10.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-hydraflow-0.2.10.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.2.10.dist-info/RECORD,,
+hydraflow-0.2.11.dist-info/METADATA,sha256=skZKh1wvBJWSLXE7iB6q2Gv_NsegcuJQjmmv8KgFg8s,4182
+hydraflow-0.2.11.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+hydraflow-0.2.11.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
+hydraflow-0.2.11.dist-info/RECORD,,