diff options
Diffstat (limited to 'docs/scripts/generate_svg.py')
| -rw-r--r-- | docs/scripts/generate_svg.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/docs/scripts/generate_svg.py b/docs/scripts/generate_svg.py index 0048160..f017b24 100644 --- a/docs/scripts/generate_svg.py +++ b/docs/scripts/generate_svg.py @@ -24,6 +24,7 @@ from unittest.mock import patch from rich.console import Console from rich.logging import RichHandler +from rich.progress import Progress from anta.cli.console import console from anta.cli.nrfu.utils import anta_progress_bar @@ -37,7 +38,7 @@ root.addHandler(r) OUTPUT_DIR = pathlib.Path(__file__).parent.parent / "imgs" -def custom_progress_bar() -> None: +def custom_progress_bar() -> Progress: """Set the console of progress_bar to main anta console. Caveat: this capture all steps of the progress bar.. @@ -56,7 +57,8 @@ if __name__ == "__main__": # stolen from https://github.com/ewels/rich-click/blob/main/src/rich_click/cli.py args = sys.argv[1:] script_name = args[0] - scripts = {script.name: script for script in entry_points().get("console_scripts")} + console_scripts = entry_points(group="console_scripts") + scripts = {script.name: script for script in console_scripts} if script_name in scripts: # A VALID SCRIPT WAS passed |