7721c80b327aaa6f5207b4babca80e4935996e42,tf_encrypted/protocol/__init__.py,,get_all_funcs,#,16

Before Change

def get_all_funcs() -> list:
    all_prot_methods = []
    all_prot_method_names = []

    protocols = [Pond(), SecureNN()]

    for protocol in protocols:
        methods = inspect.getmembers(Pond(), predicate=inspect.ismethod)
        for method in methods:
            if method[0] not in all_prot_method_names and method[0][0] is not "_":
                all_prot_method_names.append(method[0])
                all_prot_methods.append(method)

    return all_prot_methods


__all__ = [

After Change

    protocols = [Pond, SecureNN]
    for protocol in protocols:
        all_prot_method_names |= set(
            func_name
            for func_name, _ in inspect.getmembers(protocol, predicate=inspect.isfunction)
            if not func_name.startswith("_")
        )

    return all_prot_method_names

Instances