Skip to content

declearn.test_utils.list_available_frameworks

List available Vector backend frameworks.

Source code in declearn/test_utils/_vectors.py 
41
42
43
44
45
46
47
48
49
50
51
def list_available_frameworks() -> List[FrameworkType]:
    """List available Vector backend frameworks."""
    available = []
    for framework in typing.get_args(FrameworkType):
        try:
            importlib.metadata.distribution(framework)
        except importlib.metadata.PackageNotFoundError:
            pass
        else:
            available.append(framework)
    return available