Fix info under live execution table. (#267)

Author: tobiasraabe

docs/source/changes.md

@@ -5,7 +5,12 @@ chronological order. Releases follow [semantic versioning](https://semver.org/)
 releases are available on [PyPI](https://pypi.org/project/pytask) and
 [Anaconda.org](https://anaconda.org/conda-forge/pytask).
 
-## 0.2.1 - 2022-xx-xx
+## 0.2.2 - 2022-xx-xx
+
+- {pull}`267` fixes the info under the live execution table to show the total number of
+  tasks also for pytask-parallel.
+
+## 0.2.1 - 2022-04-28
 
 - {pull}`259` adds an `.svg` for profiling tasks.
 - {pull}`261` adds a config file option to sort entries in live table

docs/source/developers_guide.md

@@ -41,7 +41,7 @@ To profile pytask, you can follow this
 `git bisect`, caching, and profiling tools). We use {mod}`cProfile` with
 
 ```console
-$ python -m cProfile -o log.pstats pytask directory/with/tasks
+$ python -m cProfile -o log.pstats -m pytask directory/with/tasks
 ```
 
 The profile can be visualized with

src/_pytask/live.py

@@ -93,11 +93,14 @@ def pytask_post_parse(config: dict[str, Any]) -> None:
     config["pm"].register(live_collection, "live_collection")
 
 
-@hookimpl(tryfirst=True)
-def pytask_execute_build(session: Session) -> None:
+@hookimpl(hookwrapper=True)
+def pytask_execute(session: Session) -> Generator[None, None, None]:
     if session.config["verbose"] >= 1:
         live_execution = session.config["pm"].get_plugin("live_execution")
-        live_execution.n_tasks = len(session.tasks)
+        if live_execution:
+            live_execution.n_tasks = len(session.tasks)
+
+    yield
 
 
 @attr.s(eq=False)