def _get_deps_info()

in pyrit/show_versions.py [0:0]

• 24 lines of code
• 3 McCabe index (conditional complexity)

def _get_deps_info():
    """Overview of the installed version of main dependencies.

    This function does not import the modules to collect the version numbers
    but instead relies on standard Python package metadata.

    Returns
    -------
    deps_info: dict
        version information on relevant Python libraries
    """
    deps = sorted(
        [
            "Cython",
            "duckdb",
            "openai",
            "packaging",
            "pip",
            "numpy",
            "scipy",
            "setuptools",
            "torch",
            "transformers",
        ]
    )

    from pyrit import __version__

    deps_info = {"pyrit": __version__}

    from importlib.metadata import PackageNotFoundError, version

    for modname in deps:
        try:
            deps_info[modname] = version(modname)
        except PackageNotFoundError:
            deps_info[modname] = None
    return deps_info