From b4c2a706045911ee8d87166314de4fcb57bda563 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 8 Apr 2024 12:12:35 -0400 Subject: [PATCH] improve readability --- src/axolotl/cli/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py index 3f30a6777d..0670f64e3a 100644 --- a/src/axolotl/cli/__init__.py +++ b/src/axolotl/cli/__init__.py @@ -68,12 +68,13 @@ def print_axolotl_text_art(suffix=None): def print_dep_versions(): packages = ["accelerate", "peft", "transformers", "trl", "torch", "bitsandbytes"] + max_len = max(len(pkg) for pkg in packages) if is_main_process(): print("*" * 40) print("**** Axolotl Dependency Versions *****") for pkg in packages: version = _is_package_available(pkg, return_version=True) - print(f"{pkg: >18}: {version[1]: <15}") + print(f"{pkg: >{max_len}}: {version[1]: <15}") print("*" * 40)