def coulomb_log(*args, method="A"):
    methods = {"A": transport_A, "B": transport_B}
    tfunc = methods[method]
    return tfunc(*args)

def coulomb_log_A(*args):
    ...
    returns A

def coulomb_log_B(*args):
    ...
    returns B