def get_env_info()

in minihack/scripts/collect_env.py [0:0]

• 46 lines of code
• 5 McCabe index (conditional complexity)

def get_env_info():
    run_lambda = run
    pip_version, pip_list_output = get_pip_packages(run_lambda)

    if NLE_AVAILABLE:
        nle_version = nle.__version__
    else:
        nle_version = "N/A"

    if MINIHACK_AVAILABLE:
        minihack_version = minihack.__version__
    else:
        minihack_version = "N/A"

    if GYM_AVAILABLE:
        gym_version = gym.__version__
    else:
        gym_version = "N/A"

    if TORCH_AVAILABLE:
        torch_version_str = torch.__version__
        torch_debug_mode_str = torch.version.debug
        cuda_available_str = torch.cuda.is_available()
        cuda_version_str = torch.version.cuda
    else:
        torch_version_str = (
            torch_debug_mode_str
        ) = cuda_available_str = cuda_version_str = "N/A"

    return SystemEnv(
        minihack_version=minihack_version,
        nle_version=nle_version,
        gym_version=gym_version,
        torch_version=torch_version_str,
        is_debug_build=torch_debug_mode_str,
        python_version="{}.{}".format(
            sys.version_info[0], sys.version_info[1]
        ),
        is_cuda_available=cuda_available_str,
        cuda_compiled_version=cuda_version_str,
        cuda_runtime_version=get_running_cuda_version(run_lambda),
        nvidia_gpu_models=get_gpu_info(run_lambda),
        nvidia_driver_version=get_nvidia_driver_version(run_lambda),
        cudnn_version=get_cudnn_version(run_lambda),
        pip_version=pip_version,
        pip_packages=pip_list_output,
        conda_packages=get_conda_packages(run_lambda),
        os=get_os(run_lambda),
        gcc_version=get_gcc_version(run_lambda),
        cmake_version=get_cmake_version(run_lambda),
    )