Skip to content

Commit

Permalink
Mod: Update print_install_info function.
Browse files Browse the repository at this point in the history
  • Loading branch information
Labbeti committed Jul 6, 2023
1 parent 9365322 commit fd96562
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion src/aac_metrics/info.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import platform
import sys

from pathlib import Path
from typing import Dict

import torch
Expand All @@ -23,10 +24,16 @@ def get_install_info() -> Dict[str, str]:
}


def get_package_repository_path() -> str:
return str(Path(__file__).parent.parent.parent)


def print_install_info() -> None:
"""Print packages versions and architecture info."""
"""Show main packages versions."""
install_info = get_install_info()
pkg_repo_path = get_package_repository_path()
print(yaml.dump(install_info, sort_keys=False))
print("Package source path:", pkg_repo_path)


if __name__ == "__main__":
Expand Down

0 comments on commit fd96562

Please sign in to comment.