Check Versions

1#!/usr/bin/env python
2"""Report installed versions of key dependencies.
3
4Uses importlib.metadata.version() (works for every installed package, unlike
5the inconsistently-defined `module.__version__`).
6
7Usage:
8    python check_versions.py
9    python check_versions.py boto3 botocore transformers
10"""
11
12import sys
13from importlib.metadata import PackageNotFoundError, version
14
15DEFAULT_PACKAGES = ["boto3", "botocore", "awscli"]
16
17
18def main() -> int:
19    packages = sys.argv[1:] if len(sys.argv) > 1 else DEFAULT_PACKAGES
20    for pkg in packages:
21        try:
22            print(f"{pkg}=={version(pkg)}")
23        except PackageNotFoundError:
24            print(f"{pkg}: NOT INSTALLED")
25    return 0
26
27
28if __name__ == "__main__":
29    sys.exit(main())