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

hydraflow/progress.py

@@ -160,12 +160,11 @@ def multi_tasks_progress(
     iterables = list(iterables)
 
     with Progress(*columns, transient=transient or False, **kwargs) as progress:
-        n = len(iterables)
+        task_main = progress.add_task(main_description, total=None)
 
-        task_main = progress.add_task(main_description, total=None) if n > 1 else None
-        tasks = [
+        task_ids = [
             progress.add_task(description.format(i), start=False, total=None)
-            for i in range(n)
+            for i in range(len(iterables))
         ]
 
         total = {}
@@ -174,7 +173,7 @@ def multi_tasks_progress(
         def func(i: int) -> None:
             completed[i] = 0
             total[i] = None
-            progress.start_task(tasks[i])
+            progress.start_task(task_ids[i])
 
             for index in iterables[i]:
                 if isinstance(index, tuple):
@@ -182,21 +181,17 @@ def multi_tasks_progress(
                 else:
                     completed[i] = index + 1
 
-                progress.update(tasks[i], total=total[i], completed=completed[i])
-                if task_main is not None:
-                    if all(t is not None for t in total.values()):
-                        t = sum(total.values())
-                    else:
-                        t = None
-                    c = sum(completed.values())
-                    progress.update(task_main, total=t, completed=c)
+                progress.update(task_ids[i], total=total[i], completed=completed[i])
 
-            if transient or n > 1:
-                progress.remove_task(tasks[i])
+                if all(t is not None for t in total.values()):
+                    t = sum(total.values())
+                else:
+                    t = None
+                c = sum(completed.values())
+                progress.update(task_main, total=t, completed=c)
 
-        if n > 1:
-            it = (joblib.delayed(func)(i) for i in range(n))
-            joblib.Parallel(n_jobs, prefer="threads")(it)
+            if transient is not False:
+                progress.remove_task(task_ids[i])
 
-        else:
-            func(0)
+        it = (joblib.delayed(func)(i) for i in range(len(iterables)))
+        joblib.Parallel(n_jobs, prefer="threads")(it)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.2.9
+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.9.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=UIIKlweji3L0uRi4hZ_DrtRcnayHPlsMoug7hVEKq8k,6753
+hydraflow/progress.py,sha256=uaxS9UnwsikOPB_Ho-5mXdBYz3Zedj-SL9FzT094wjY,6623
 hydraflow/run_collection.py,sha256=V5lGdGHYgsSpBOYGaVEL1mpKJvdiEshBL0KmmZ8qeZo,29161
-hydraflow-0.2.9.dist-info/METADATA,sha256=ZjJQz_4MogGkcs16dOwnsp_J0icg9ypgQdXOYxVdxJg,4181
-hydraflow-0.2.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-hydraflow-0.2.9.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.2.9.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,,